\begin{equation}
\label{eq:dsm_loss}
    \mathcal{L}(\theta) = \mathbb{E}[\lambda_{t} \lVert \nabla \log p_{t | 0}(\mathbf{X}^{(t)} | \mathbf{X}^{(0)}) - s_{\theta}(t, \mathbf{X}^{(t)}) \rVert^{2}],
\end{equation}
