用信息论优化磁共振成像采样,同时实现不确定性和任务自适应。
Information-Theoretic Optimization for Task-Adapted Compressed Sensing Magnetic Resonance Imaging
- 基于互信息最大化,统一优化采样、重建与临床任务
- 在Dice分数上优于传统方法,且后验分布匹配度更高
- 支持隐私保护场景下的任务执行,适用于多种临床需求
面向下游临床任务的自适应压缩感知磁共振成像(CS-MRI)正兴起,以在远低于奈奎斯特采样要求的情况下减少k空间测量。然而现有方法存在诊断不确定性问题,且无法在端到端优化中实现采样自适应。为此,本文首次从信息论角度提出任务自适应CS-MRI框架,通过最大化欠采样k空间测量与临床任务间的互信息,实现不确定性概率推断,并支持任意采样率和多样化临床应用。我们采用摊销优化并构建可计算的变分界来联合优化采样、重建与任务推理模型,仅需一个端到端训练模型即可灵活控制采样比例。该框架统一处理两种临床场景:一是重建作为辅助以提升任务性能;二是抑制重建以保护隐私。大规模MRI数据集上的实验表明,该方法在标准指标如Dice上表现优异,且在广义能量距离(GED)上更接近真实后验分布。
原文摘要 · Abstract (English)
Task-adapted compressed sensing magnetic resonance imaging (CS-MRI) is emerging to address the specific demands of downstream clinical tasks with significantly fewer k-space measurements than required by Nyquist sampling. However, existing task-adapted CS-MRI methods suffer from the uncertainty problem for medical diagnosis and cannot achieve adaptive sampling in end-to-end optimization with reconstruction or clinical tasks. To address these limitations, we propose the first task-adapted CS-MRI from the information-theoretic perspective to simultaneously achieve probabilistic inference for uncertainty prediction and adapt to arbitrary sampling ratios and versatile clinical applications. Specifically, we formalize the task-adapted CS-MRI optimization problem by maximizing the mutual information between undersampled k-space measurements and clinical tasks to enable probabilistic inference for addressing the uncertainty problem. We leverage amortized optimization and construct tractable variational bounds for mutual information to jointly optimize sampling, reconstruction, and task-inference models, which enables flexible sampling ratio control using a single end-to-end trained model. Furthermore, the proposed framework addresses two kinds of distinct clinical scenarios within a unified approach, i.e., i) joint task and reconstruction, where reconstruction serves as an auxiliary process to enhance task performance; and ii) task implementation with suppressed reconstruction, applicable for privacy protection. Extensive experiments on large-scale MRI datasets demonstrate that the proposed framework achieves highly competitive performance on standard metrics like Dice compared to deterministic counterpart but provides better distribution matching to the ground-truth posterior distribution as measured by the generalized energy distance (GED).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。