Multi-task Neural Diffusion Processes
Multi-task neural diffusion processes extend neural diffusion models to handle multiple related tasks by adding a task encoder. This encoder conditions the model on context observations, allowing for task-conditioned probabilistic regression and few-shot adaptation across functions. The method improves predictive accuracy and uncertainty calibration, especially when context data is available.