Adaptive Inference-Time Scaling via Cyclic Diffusion Search
Abstract
Diffusion models have demonstrated strong generative capabilities across domains ranging from image synthesis to complex reasoning tasks. However, most inference-time scaling methods rely on fixed denoising schedules, limiting their ability to allocate computation based on instance difficulty or task-specific demands adaptively. We introduce the challenge of adaptive inference-time scaling—dynamically adjusting computational effort during inference—and propose Adaptive Bi-directional Cyclic Diffusion (ABCD), a flexible, search-based inference framework. ABCD refines outputs through bi-directional diffusion cycles while adaptively controlling exploration depth and termination. It comprises three components: Cyclic Diffusion Search, Automatic Exploration-Exploitation Balancing, and Adaptive Thinking Time. Experiments show that ABCD improves performance across diverse tasks while maintaining computational efficiency.
Cite
Text
Lee et al. "Adaptive Inference-Time Scaling via Cyclic Diffusion Search." Advances in Neural Information Processing Systems, 2025.Markdown
[Lee et al. "Adaptive Inference-Time Scaling via Cyclic Diffusion Search." Advances in Neural Information Processing Systems, 2025.](https://mlanthology.org/neurips/2025/lee2025neurips-adaptive-a/)BibTeX
@inproceedings{lee2025neurips-adaptive-a,
title = {{Adaptive Inference-Time Scaling via Cyclic Diffusion Search}},
author = {Lee, Gyubin and Truong, Bao N Nguyen and Yoon, Jaesik and Lee, Dongwoo and Kim, Minsu and Bengio, Yoshua and Ahn, Sungjin},
booktitle = {Advances in Neural Information Processing Systems},
year = {2025},
url = {https://mlanthology.org/neurips/2025/lee2025neurips-adaptive-a/}
}