TROI通过稀疏体素选择提升跨被试fMRI视觉解码性能
TROI: Cross-Subject Pretraining with Sparse Voxel Selection for Enhanced fMRI Visual Decoding
- 提出两阶段数据驱动体素选择方法,自动定位有效脑区
- 在小样本下实现比MindEye2更高的图像重建与检索准确率
- 适合资源有限的跨被试脑信号解码研究者使用
fMRI视觉解码旨在从视觉刺激引发的大脑信号中还原原始图像。传统方法依赖人工标注的感兴趣区域(ROIs)选取脑体素,但这些区域常含冗余信息和噪声,影响解码效果。此外,缺乏自动化ROI标注方法限制了fMRI视觉解码技术的实际应用,尤其在新被试上。本文提出TROI(可训练兴趣区),一种针对跨被试fMRI解码任务的两阶段数据驱动ROI标注方法,尤其适用于样本量有限的情况。TROI利用数据集中的标注ROI预训练一个图像解码主干网络,使新被试能高效优化输入层而无需从头训练整个模型。第一阶段结合稀疏掩码训练与低通滤波,快速生成体素掩码并确定输入层维度;第二阶段采用学习率重绕策略微调输入层以适应下游任务。在与基线方法相同的小样本数据集上进行的脑视觉检索与重建实验表明,本方法的体素选择策略优于当前最优方法MindEye2所用的人工标注ROI掩码。
原文摘要 · Abstract (English)
fMRI (functional Magnetic Resonance Imaging) visual decoding involves decoding the original image from brain signals elicited by visual stimuli. This often relies on manually labeled ROIs (Regions of Interest) to select brain voxels. However, these ROIs can contain redundant information and noise, reducing decoding performance. Additionally, the lack of automated ROI labeling methods hinders the practical application of fMRI visual decoding technology, especially for new subjects. This work presents TROI (Trainable Region of Interest), a novel two-stage, data-driven ROI labeling method for cross-subject fMRI decoding tasks, particularly when subject samples are limited. TROI leverages labeled ROIs in the dataset to pretrain an image decoding backbone on a cross-subject dataset, enabling efficient optimization of the input layer for new subjects without retraining the entire model from scratch. In the first stage, we introduce a voxel selection method that combines sparse mask training and low-pass filtering to quickly generate the voxel mask and determine input layer dimensions. In the second stage, we apply a learning rate rewinding strategy to fine-tune the input layer for downstream tasks. Experimental results on the same small sample dataset as the baseline method for brain visual retrieval and reconstruction tasks show that our voxel selection method surpasses the state-of-the-art method MindEye2 with an annotated ROI mask.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。