theodpzz/ct-ssg

[Under review] Structured Spectral Graph Representation Learning for Multi-label Abnormality Analysis from 3D CT Scans 🩺👨🏻‍⚕️

✅ Official implementation of the paper "Structured Spectral Graph Representation Learning for Multi-label Abnormality Analysis from 3D CT Scans".

📄 Under review, preprint: arXiv preprint.

⚡️ Source code available at https://github.com/theodpzz/ct-ssg.

🔥 Available resources

model_state_dict.pt: Model weights for CT-SSG trained on the CT-RATE training set.

thresholds.json: Per-abnormality classification thresholds optimized on our internal CT-RATE validation set. The official CT-RATE test set was not used during threshold optimization to preserve unbiased evaluation.

🤝🏻 Acknowledgment

We thank contributors from the CT-RATE dataset available at https://huggingface.co/datasets/ibrahimhamamci/CT-RATE, from the Rad-ChestCT dataset available at https://zenodo.org/records/6406114 and from the Merlin Abdominal CT dataset available at https://stanfordaimi.azurewebsites.net/categories/datasets?domain=BODY.

📎Citation

If you find this repository useful for your work, we would appreciate the following citation:

@article{dipiazza_2026_ctssg,
        title = {Structured Spectral Graph Representation Learning for Multi-label Abnormality Analysis from 3D CT Scan},
        author = {Di Piazza, Theo and Lazarus, Carole and Nempont, Olivier and Boussel, Loic},
        year = {2026},
        note = {Preprint, under review},
}