HyperAIHyperAI

Command Palette

Search for a command to run...

脳符号化のためのパラメータ効率的ファインチューニングのスケーリング

Josh Ying Guanglong Sun Nolan Ho Kieran Liu Rui Yang Andrew Chen Nikolaus Kriegeskorte Pony Ma Hossein Adeli

概要

脳符号化モデルは、刺激に対する神経応答の予測を目的とする。最も広く用いられている手法では、凍結した事前学習済みバックボーンから読み出し層を適合させる。神経データはコストが高く限られているため、神経データを用いて大規模モデルをゼロから学習したり、タスク事前学習後に完全にファインチューニングしたりすることは困難であることが示されている。我々は、パラメータ効率的ファインチューニング(PEFT)により、少数の学習可能パラメータで、脳符号化のために非常に大規模な事前学習済みバックボーンを効果的に適応できることを示す。これを計算上実用的にするため、バックボーンの因果構造を利用して、スライディングウィンドウではなく1回の順伝播で複数の時点を予測し、特徴抽出コストを大幅に削減する。これにより、最大104Bパラメータのマルチモーダルバックボーンを効率的にファインチューニングでき、我々の知る限り、これは脳符号化に適応された最大のモデルである。低ランクアダプタ(LoRA)と軽量な読み出し層を用いて学習する我々の手法は、9つの多様なマルチモーダルバックボーンにわたって脳符号化性能を一貫して向上させる。先行する最先端(SoTA)と同じQwen3-Omniバックボーン上で、我々の手法BrainLoRAは、Algonauts 2025 Challengeにおいて公表済みの最高性能を達成し、学習可能パラメータ数は約5%にすぎない。さらに、適応されたマルチモーダルバックボーンが解釈可能性を保つことも示す。モダリティ帰属は、先行研究が別々の単一モダリティエンコーダから構成的に得ることが多い標準的な感覚トポグラフィーを回復し、神経データへの適応はバックボーン内部での音声入力とクロスモーダル混合の寄与を増加させる。適応モデルの数を拡張するため、共有バックボーン上で複数のLoRAを同時学習することを高速化するマルチLoRAカーネルを開発する。これにより、200以上の適応モデルからなるアンサンブルが可能となり、本稿執筆時点で公開Algonautsポストチャレンジリーダーボードにおいて1位となっている。以上の結果は、PEFTが脳符号化のための効率的かつ強力なパラダイムであることを示している。

One-sentence Summary

Researchers from Columbia University et al. propose BrainLoRA, a parameter-efficient fine-tuning method that adapts multimodal backbones up to 104B parameters for brain encoding via low-rank adapters, lightweight readouts, and single-pass causal multi-timepoint prediction, achieving state-of-the-art Algonauts 2025 performance with about 5% trainable parameters and ranking first with a 200+ model ensemble.

Key Contributions

  • The paper presents BrainLoRA, a parameter-efficient fine-tuning method that adapts large pretrained multimodal backbones for brain encoding using low-rank adapters and a lightweight readout. It improves brain encoding across nine multimodal backbones and achieves the best published Algonauts 2025 Challenge performance on the same Qwen3-Omni backbone used by prior work while using roughly 5% of the trainable parameters.
  • The work makes large-scale adaptation computationally practical by exploiting the backbone's causal structure for single-pass multi-timepoint feature extraction, reducing feature-extraction cost by 350x compared with prior state of the art, and by introducing a multi-LoRA kernel for batched adapter training. This supports adapting backbones up to 104B parameters and training an ensemble of more than 200 models that ranks first on the public Algonauts post-challenge leaderboard.
  • The study shows that adapted multimodal backbones remain interpretable: modality attribution recovers canonical sensory topography normally obtained by combining separate unimodal encoders, and adaptation increases audio input contributions and cross-modal mixing compared with frozen backbones.

Introduction

Brain encoding models predict neural responses to stimuli and are usually built by freezing a large pretrained model as a feature extractor and training only a readout on neural data. This approach benefits from strong pretrained priors but cannot specialize the backbone to brain data, and adapting backbones is difficult because neural recordings are costly and scarce. Prior adaptation results have been mixed: training smaller models from scratch does not reach state-of-the-art whole-brain performance, full fine-tuning can underperform freezing, and low-rank adaptation had only been tested on smaller vision models. The authors address this gap by treating parameter-efficient fine-tuning as a scalable framework for brain encoding. They introduce BrainLoRA, which pairs LoRA with a lightweight readout on large multimodal backbones, and make adaptation practical through single-pass feature extraction that reduces cost by roughly 350 times. BrainLoRA improves over frozen backbones across nine multimodal models up to 104B parameters and sets a new state of the art on the Algonauts 2025 whole-brain fMRI benchmark while using about 5 percent of the trainable parameters of the prior best method.

Method

The authors formalize brain encoding as a mapping from a multimodal stimulus XXX to the evoked neural response YYY. The model consists of a pretrained backbone fθf_\thetafθ​ followed by a readout gϕg_\phigϕ​, trained by minimizing

Y^=gϕ(fθ(X)),min⁡θ∈Θ,ϕL(Y,Y^),\hat{Y} = g_\phi\big(f_\theta(X)\big), \qquad \min_{\theta \in \Theta, \phi} \mathcal{L}\big(Y, \hat{Y}\big),Y^=gϕ​(fθ​(X)),θ∈Θ,ϕmin​L(Y,Y^),

where L\mathcal{L}L is a task-appropriate loss such as squared error or negative correlation, and gϕg_\phigϕ​ also handles temporal alignment between the stimulus representation and the neural sampling rate.

Instead of full fine-tuning, the method constrains backbone adaptation to a low-dimensional parameterization. Given pretrained weights θ0\theta_0θ0​, the optimization set satisfies

Θfrozen={θ0}⊂ΘPEFT={θ0+Δ(ψ):ψ∈Rk}⊂Θfull=Rd,\Theta_{\text{frozen}} = \{\theta_0\} \subset \Theta_{\text{PEFT}} = \{\theta_0 + \Delta(\psi): \psi \in \mathbb{R}^{k}\} \subset \Theta_{\text{full}} = \mathbb{R}^{d},Θfrozen​={θ0​}⊂ΘPEFT​={θ0​+Δ(ψ):ψ∈Rk}⊂Θfull​=Rd,

with k≪dk \ll dk≪d and Δ(0)=0\Delta(0) = 0Δ(0)=0. This formulation is architecture-agnostic, and LoRA, adapter layers, and prompt tuning are all instances of Δ\DeltaΔ. For the main architecture based on Qwen3-Omni, the authors insert rank-16 LoRA into attention modules through layer 32 and truncate the decoder at that layer, introducing 8.91 million trainable parameters. The adapters and readout are jointly optimized for one epoch.

The feature extraction pipeline replaces the overlapping sliding-window protocol used by earlier methods. Because the backbone is causally masked, a 149-second window, corresponding to 100 fMRI TRs, is encoded with a single forward pass, and representations for all timesteps are extracted simultaneously. A 60-second context lead is prepended to each window to recover the temporal context lost at window boundaries. This single-pass design reduces feature extraction cost by roughly 350 times and makes fine-tuning through the backbone computationally affordable. The authors also verify that the locally bidirectional modality encoders, which operate over short audio and visual blocks before the decoder, introduce only small numerical differences rather than meaningful future stimulus leakage.

For the readout, backbone features are first pooled into one vector per modality at each 2 Hz timestep and aggregated into eight log-spaced temporal-lag blocks aligned with the fMRI sampling rate. The linear readout pools roughly 19.3k tokens to 894 vectors per input window and then applies a final linear projection to 1,000 brain parcels, with 52.3 million trainable parameters. As an alternative, per-TR cross-attention over an expanded set of about 3.2k tokens replaces the fixed temporal-lag aggregation, adding only 0.2 million parameters.

After joint optimization with SGD, the linear readout WWW is re-estimated in closed form. Forward passes accumulate subject-specific Gram matrices, after which WWW is solved using parcel-wise ridge regression. This closed-form re-fit provides an additional improvement beyond the SGD-trained readout.

Experiment

Using mean Pearson correlation across four subjects on an audiovisual fMRI benchmark, the experiments evaluate parameter-efficient adaptation across nine multimodal backbones, compare small adapters against large readouts, ablate model components and input design, interpret modality contributions, and study large adapter ensembles. The findings show that PEFT consistently improves brain encoding over frozen backbones, with lightweight LoRA or cross-attention adapters outperforming much larger frozen readouts and setting a new state of the art, while efficient single-pass inputs recover lost performance after adaptation. Feature attribution recovers canonical cortical organization but adaptation shifts modality weighting toward audio and increases cross-modal mixing. Large ensembles of independently adapted models further improve accuracy, with useful diversity arising mainly from architectural changes rather than seed or training variations.

BrainLoRA sets a new state of the art on the Algonauts 2025 benchmark by adapting a frozen backbone with lightweight LoRA modules instead of training a large readout. A LoRA plus linear adapter with a small fraction of MIRAGE's trainable parameters already outperforms the previous single-model state of the art on held-out ID and OOD test sets, and a cross-attention variant improves further. Full fine-tuning of the feature extraction stack does not beat LoRA, while ensembling many independently adapted models yields the highest public leaderboard score. A LoRA plus linear adapter trains roughly 5 percent of the prior MIRAGE model's parameters and already surpasses it on both ID and OOD held-out scores. A cross-attention readout further improves single-model performance, exceeding the published 15-model MIRAGE ensemble on ID and matching it on OOD. Full fine-tuning with hundreds of times more trainable parameters performs on par with or slightly below LoRA, and a large ensemble of adapters achieves the top public leaderboard result.

The ablation shows that adapting the frozen backbone with LoRA yields the largest performance gain, and the closed-form re-fit adds a smaller but consistent improvement. Alternative readout and rank choices produce modest differences, with cross-attention showing the highest score among the readout variants. Removing audio causes a substantial drop, indicating its importance to the encoding model. LoRA adaptation provides the largest single improvement over the frozen-backbone baseline. Removing the closed-form re-fit lowers the adapted model score, confirming its consistent contribution. Rank-32 LoRA and gated-MLP readout perform close to the main configuration, while cross-attention readout scores slightly higher. No audio leads to a substantial performance drop relative to the main configuration.

These experiments evaluate BrainLoRA on the Algonauts 2025 benchmark, using a frozen backbone adapted with lightweight LoRA modules and different readouts. LoRA adaptation provides the largest gain over the frozen baseline, while the closed-form re-fit adds a smaller but consistent improvement and removing audio causes a substantial performance drop. A LoRA plus linear adapter already surpasses the prior MIRAGE model with far fewer parameters, and a cross-attention readout improves single-model performance further. Full fine-tuning does not outperform LoRA, and ensembling adapted models yields the best leaderboard result.


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています