acceptodds
Under review as a conference paper at ICLR 2027

Reconstruct What the Task Needs: Progressive Task-Prior Information Bottlenecks for Accelerated MRI

Abstract

Magnetic resonance imaging (MRI) reconstruction is increasingly integrated with downstream tasks to preserve task-relevant structures. However, in multi-stage unrolled networks, stage-wise task feedback can entangle newly emerging task information with redundant and task-irrelevant content. We propose Purified Task Prior Guided MRI (PTP-MRI), which represents each stage update as a history-conditioned stochastic task increment. A variational lower bound encourages the retention of newly emerging task information, while a conditional variational upper bound limits task-irrelevant residual information. The resulting constrained information objective is optimized through Lagrangian duality with an adaptive dual variable. The purified task states are progressively accumulated and decoded into spatial priors that guide reconstruction according to task relevance. Experiments on SKM-TEA at acceleration and AHEAD at acceleration show that PTP-MRI improves downstream segmentation while maintaining reasonable reconstruction fidelity. Compared with the competing baseline for each metric, PTP-MRI achieves Dice/mIoU gains of 0.78/1.28 percentage points on SKM-TEA and 0.87/1.71 percentage points on AHEAD.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.