4 papers
From Atoms to Entropy: Optimal Noise Allocation for Diffusion Training in the Convex Regime
Luca Ambrogioni, Giulio Franzese, Alberto Foresti +7
How should a diffusion model decide which noise levels to train on, and how much? Despite the importance of this choice, current noise schedules are based largely on heuristics or…
Noise Scheduling as Information-Guided Allocation in Diffusion Training
Gabriel Raya, Bac Nguyen, Georgios Batzolis +6
We introduce InfoNoise, an online adaptive noise schedule for diffusion training that reallocates optimization effort toward noise levels where denoising is most informative. Toget…
Adaptive Knowledge Transferring with Switching Dual-Student Framework for Semi-Supervised Medical Image Segmentation
Hoang-Thien Nguyen, Thanh-Huy Nguyen, Ba-Thinh Lam +6
Teacher-student frameworks have emerged as a leading approach in semi-supervised medical image segmentation, demonstrating strong performance across various tasks. However, the lea…
SAFT: Towards Out-of-Distribution Generalization in Fine-Tuning
Bac Nguyen, Stefan Uhlich, Fabien Cardinaux +3
Handling distribution shifts from training data, known as out-of-distribution (OOD) generalization, poses a significant challenge in the field of machine learning. While a pre-trai…