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/}
}