SynthBA: Brain Age Estimation for Any Brain MRI

Note

SynthBA has been peer-reviewed and accepted at IEEE MetroXRAINE 2024.

SynthBA (Synthetic Brain Age) is a deep learning model that predicts the biological age of the brain (i.e., brain age) from MRI scans. Thanks to domain randomization, it works reliably on scans of arbitrary contrast and resolution without needing retraining or harmonization.

This repository contains the official pip-installable package for SynthBA, which automatically handles preprocessing (skull stripping and alignment) and model inference.

🚀 Using synthba (Command-Line)

1. Installation

You can install SynthBA directly from PyPI:

pip install synthba

For the latest development version, you can install directly from this repository:

pip install git+https://github.com/lemuelpuglisi/synthba.git

2. Running Prediction

Once installed, you can use the synthba command in your terminal. All models and alignment templates are downloaded automatically from Hugging Face Hub on the first run.

The tool can accept a single scan, a folder of scans, or a CSV file as input.

Example 1: Running on a single scan

synthba -i /path/to/my_scan.nii.gz -o /path/to/output --device cuda

Example 2: Running on a directory of scans

This will process all .nii and .nii.gz files in the folder. We recommend using a larger batch size for faster processing if you have enough VRAM/RAM.

synthba -i /path/to/scans_folder -o /path/to/output --device cuda --batch_size 4

Example 3: Running on a CSV file

The CSV file should contain one absolute path to a NIfTI scan per line.

synthba -i /path/to/my_scan_list.csv -o /path/to/output --device cuda --batch_size 4

3. Output

The tool will generate two main outputs in your specified output directory:

1. synthba_predictions.csv: A CSV file containing the paths to the processed scans and their predicted brain age.
2. preprocessed/: A new folder containing the preprocessed (skull-stripped and aligned) scans. This folder is only created if preprocessing is enabled (the default).

4. Command-Line Options

Here is a full list of available options:

Argument	Flag	Description
Input Path	-i, --input	(Required) Path to a single NIfTI scan, a directory of scans, or a .csv file listing scan paths.
Output Directory	-o, --output_dir	(Required) Path to a directory where results (synthba_predictions.csv) and preprocessed scans will be saved.
Model Type	-m, --model_type	The model variant to use. g (gaussian prior, best model) or u (uniform prior). g is recommended.
Template	-t, --template	The template for alignment, based on your scan's MR weighting. t1 (default) or t2.
Device	-d, --device	The device to run on. cuda or cpu. If not set, it auto-detects CUDA.
Batch Size	-b, --batch_size	Number of scans to process in a single batch (default: 1).
Skip Preprocessing	--skip-prep	Use this flag if your scans are already skull-stripped and registered to MNI152 1mm space.
Checkpoint	-c, --checkpoint	Path to a custom, local model checkpoint (.pth) file. If not provided, the correct model is downloaded from Hugging Face.

🐍 Using the Python Interface

You can also import and use the SynthBA class directly in your own Python scripts for more complex workflows.

Initialization

First, import and initialize the SynthBA class.

from synthba import SynthBA

# Initialize the model
# This will download the model checkpoint and templates if not already cached
sba = SynthBA(
    device='cuda',  # or 'cpu'
    model_type='g'  # 'g' (gaussian) or 'u' (uniform)
)

run(): Predicting a single, loaded scan

The run() method is for a single nibabel image object.

import nibabel as nib

# Load your scan as a nibabel object
scan_nii = nib.load('path/to/T1w_scan.nii.gz')

# Run prediction
# preprocess=True will apply skull-stripping and alignment
# mr_weighting='t1' tells the aligner to use the T1 template
brain_age = sba.run(
    scan=scan_nii, 
    preprocess=True, 
    mr_weighting='t1'
)

print(f"Predicted Brain Age: {brain_age:.2f} years")

run_multiple(): Predicting a list of scan paths

The run_multiple() method is designed for batch processing a list of file paths. It returns a pandas.DataFrame.

Note: When preprocess=True, you must provide preprocess_outdir to specify where the processed scans should be saved.

scan_paths = [
    '/data/sub-001/anat.nii.gz',
    '/data/sub-002/anat.nii.gz',
    '/data/sub-003/anat.nii.gz'
]

# Run prediction on the list of paths
results_df = sba.run_multiple(
    input_list=scan_paths,
    batch_size=2,
    preprocess=True,
    preprocess_outdir='/path/to/save/preprocessed_files/',
    mr_weighting='t1'
)

print(results_df)
#
#                          path       pred
# 0  /data/sub-001/anat.nii.gz  34.567890
# 1  /data/sub-002/anat.nii.gz  45.123456
# 2  /data/sub-003/anat.nii.gz  28.987654

---

🐳 Legacy Docker Usage (Old Version)

For users who prefer the original Docker-based version, you can still pull and run the Docker image from DockerHub.

docker run --rm --gpus all \
    -v /path/to/inputs:/home/inputs \
    -v /path/to/outputs:/home/outputs \
    lemuelpuglisi/synthba:latest

Warning

The Docker version and the pip package are not the same. The Docker image may use older dependencies and does not support all features of the pip package (like CSV input or automatic device detection).

---

❓ FAQ & Troubleshooting

1. Error: Killed (when using Docker)

   • Cause: The Docker container ran out of memory. The preprocessing (SynthStrip) and inference steps can be RAM-intensive.
   • Solution: If using Docker Desktop (Mac or Windows), go to Settings > Resources and increase the Memory allocated to Docker (e.g., to 16GB or more).

2. Error: Process crashes or is very slow during preprocessing.

   • Cause: The skull-stripping model (SynthStrip) conforms the image to a 1mm space, which can consume a large amount of RAM (16GB+ is recommended).
   • Solution: If you are processing many subjects, use a smaller --batch_size (e.g., -b 1). If it still fails, you may need to run on a machine with more RAM.

3. Error (Python): Exception: Please specify where to store the preprocessing output with preprocess_outdir

   • Cause: You called sba.run_multiple(..., preprocess=True) without specifying the preprocess_outdir argument.
   • Solution: You must provide a path to a directory where the processed files can be saved, e.g., sba.run_multiple(..., preprocess_outdir='./preprocessed'). The command-line tool handles this automatically.

4. Error (PyTorch/macOS): RuntimeError: 'aten::max_pool3d_with_indices' is not implemented for 'mps'

   • Cause: This error occurs when running the model on the MPS (Metal Performance Shaders) device on macOS (Apple Silicon). The required operation was not fully implemented in the stable version of PyTorch.
   • Solution: This bug has been addressed in PyTorch's development branch (see here) and will be included in the next stable release. To resolve it immediately, you must install the PyTorch nightly build from the PyTorch website. This will enable the missing MPS operation and allow the model to run on your Mac!

🙏 Credits

SynthBA is built on top of several outstanding open-source projects:

• SynthStrip: For robust skull stripping.
• ANTsPy: For affine registration to the MNI template.
• MONAI: For data transformations and the DenseNet architecture.
• PyTorch: As the core deep learning framework.
• Hugging Face Hub: For model and template hosting.

📜 Citing

If you use SynthBA for your research, please cite our paper:

@inproceedings{puglisi2024synthba,
  title={SynthBA: Reliable Brain Age Estimation Across Multiple MRI Sequences and Resolutions},
  author={Puglisi, Lemuel and Rondinella, Alessia and De Meo, Linda and Guarnera, Francesco and Battiato, Sebastiano and Rav{\`\i}, Daniele},
  booktitle={2024 IEEE International Conference on Metrology for eXtended Reality, Artificial Intelligence and Neural Engineering (MetroXRAINE)},
  pages={559--564},
  year={2024},
  organization={IEEE}
}}

And the follow-up paper demonstrating SynthBA's efficacy on low-field MRI scans:

@article{biondo2025brain,
  title={Brain-age in ultra-low-field MRI: how well does it work?},
  author={Biondo, Francesca and Bennallick, Carly and Martin, Sophie A and Puglisi, Lemuel and Booth, Thomas C and Wood, David A and Iglesias, Juan E and V{\'a}{\v{s}}a, Franti{\v{s}}ek and Cole, James H},
  journal={medRxiv},
  pages={2025--10},
  year={2025},
  publisher={Cold Spring Harbor Laboratory Press}
}