面向自回归MRI重建的"下一加速尺度预测"
阅读原文· arxiv.org针对高加速欠采样下MRI重建因模糊性导致高频细节丢失的问题,该研究将重建过程移至离散多尺度潜空间,并构建为自回归的“下一加速尺度预测”任务。方法利用视觉自回归建模中有效的离散先验,将解空间约束在紧凑的码本token序列中,从而即使从极度稀疏的测量中也能实现清晰重建。该框架自然适配大语言模型的后训练技术,并引入了在线策略蒸馏,利用教师模型在推理时不可用的特权上下文(完全采样数据)监督学生模型。在fastMRI基准测试的多种极端欠采样模式下,该方法均展现出改进的重建效果。
MRI reconstruction is an inherently ill-posed inverse problem, since incomplete measurements admit many plausible solutions. This ambiguity becomes more severe under high acceleration, where pixel-domain continuous predictors tend to average over feasible reconstructions and suppress high-frequency anatomy. We address this limitation by moving reconstruction to discrete multi-scale latent space and posing it as autoregressive next-acceleration-scale prediction. Leveraging discrete priors proven effective in visual autoregressive modeling, our method restricts the solution to compact sequences of codebook tokens, enabling sharp reconstructions even from extremely sparse measurements. This discrete autoregressive formulation also aligns naturally with modern large language model post-training techniques. Building on this observation, we introduce on-policy privileged information distillation for visual autoregressive modeling, where a teacher is provided training only privileged context that is unavailable at inference, in our case fully sampled acquisitions, and supervises a student trained on its own rollouts, leading to consistent reconstruction gains. Through extensive experiments on the fastMRI benchmark, we show that our approach delivers improved reconstruction performance across diverse sampling patterns under extreme undersampling. Project website is https://yilmazkorkmaz1.github.io/discrete-mri-reconstruction-opd/{here}.