In stochastic optimal control and conditional generative modelling, a central computational task is to modify a reference diffusion process to maximise a given terminal-time reward. Most existing methods require this reward to be differentiable, using gradients to steer the diffusion towards favourable outcomes. However, in many practical settings, like diffusion bridges, the reward is singular, taking an infinite value if the target is hit and zero otherwise. We introduce a novel framework, based on Malliavin calculus and path-space integration by parts, that enables the development of methods robust to such singular rewards. This allows our approach to handle a broad range of applications, including classification, diffusion bridges, and conditioning without the need for artificial observational noise. We demonstrate that our approach offers stable and reliable training, outperforming existing techniques.
View on arXiv@article{pidstrigach2025_2504.03461,
title={ Conditioning Diffusions Using Malliavin Calculus },
author={ Jakiw Pidstrigach and Elizabeth Baker and Carles Domingo-Enrich and George Deligiannidis and Nikolas Nüsken },
journal={arXiv preprint arXiv:2504.03461},
year={ 2025 }
}