用小步生成高质量3D脑部MRI,助力年龄预测模型更公平
FlowLet: Conditional 3D Brain MRI Synthesis using Wavelet Flow Matching
- 在可逆小波域做流匹配,直接生成带年龄条件的3D MRI
- 仅需5~10步采样即生成高保真体积数据,速度快于扩散模型
- 生成数据能提升少数年龄群体的预测性能,结构保持准确
脑部磁共振成像(MRI)在神经发育、衰老和疾病研究中至关重要。其关键应用之一是脑龄预测(BAP),即通过MRI数据估算个体的生物脑龄。高效BAP模型依赖大规模、多样且年龄均衡的数据集,但现有3D MRI数据集存在人口学偏差,限制了公平性和泛化能力。新数据采集成本高且受伦理制约,催生了生成式数据增强需求。当前方法多基于潜空间扩散模型,虽缓解了体积数据的内存压力,但推理慢、易引入伪影,且极少支持年龄条件,影响BAP效果。本文提出FlowLet,一种基于可逆3D小波域流匹配的条件生成框架,可生成带年龄条件的3D MRI,避免重建伪影并降低计算开销。实验表明,FlowLet仅需5~10步采样即可生成高保真体积数据。用其生成的数据训练BAP模型,显著提升少数年龄群体的表现,区域分析证实解剖结构得以良好保留。
原文摘要 · Abstract (English)
Brain Magnetic Resonance Imaging (MRI) plays a central role in studying neurological development, aging, and diseases. One key application is Brain Age Prediction (BAP), which estimates an individual's biological brain age from MRI data. Effective BAP models require large, diverse, and age-balanced datasets, whereas existing 3D MRI datasets are demographically skewed, limiting fairness and generalizability. Acquiring new data is costly and ethically constrained, motivating generative data augmentation. Current generative methods are often based on latent diffusion models, which operate in learned low dimensional latent spaces to address the memory demands of volumetric MRI data. However, these methods are typically slow at inference, may introduce artifacts due to latent compression, and are rarely conditioned on age, thereby affecting the BAP performance. In this work, we propose FlowLet, a conditional generative framework that synthesizes age-conditioned 3D MRIs by leveraging flow matching within an invertible 3D wavelet domain, helping to avoid reconstruction artifacts and reducing computational demands. Experiments show that FlowLet generates high-fidelity volumes with few sampling steps. Training BAP models with data generated by FlowLet improves performance for underrepresented age groups, and region-based analysis confirms preservation of anatomical structures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。