🧠 Optimal Transport for Brain-Image Alignment: Unveiling Redundancy and Synergy in Neural Information Processing [ICCV 2025]

Official Implementation of ICCV 2025 Paper

Usage

Data preparation, and training procedures, please refer to https://github.com/weihaox/UMBRAE

All you need to do is just replace MSE Loss with OT Loss.

Name		Name	Last commit message	Last commit date
Latest commit History 26 Commits
alignment		alignment
model_weights		model_weights
test_images		test_images
README.md		README.md
environment.yaml		environment.yaml
inference.py		inference.py
inference_adaptation.py		inference_adaptation.py
inference_adaptation_rec.py		inference_adaptation_rec.py
inference_rec.py		inference_rec.py
model.py		model.py
perceiver.py		perceiver.py
requirements.txt		requirements.txt
train.py		train.py
train_adaptation.py		train_adaptation.py
train_brainx.py		train_brainx.py
train_brainx_ot.py		train_brainx_ot.py
train_ot.py		train_ot.py
train_ot_heatmap.py		train_ot_heatmap.py
utils.py		utils.py