Downloads · 30 days
0
lavsendahal/janus
janus is a machine learning model from lavsendahal. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Multi-label abdominal disease classification from 3-D CT scans. DINOv3 backbone with four ablation variants: GAP, Masked Attention, Scalar Fusion, and Gated Fusion.
Downloads · 30 days
0
Access
Public
Updated May 9, 2026
Repo size
1.4 GB
Likes
0
Public
Click a slice to open those files.
.safetensors1.4 GB · 100%
From the Hugging Face model README
Multi-label abdominal disease classification from 3-D CT scans. DINOv3 backbone with four ablation variants: GAP, Masked Attention, Scalar Fusion, and Gated Fusion.
| Variant | Folder | Macro AUROC | Needs seg masks | Needs radiomics |
|---|---|---|---|---|
| GAP (baseline) | gap/ | 0.8488 | — | — |
| Masked Attention | masked-attn/ | 0.8667 | ✓ | — |
| Scalar Fusion | scalar-fusion/ | 0.8332 | ✓ | ✓ |
| Gated Fusion | gated-fusion/ | 0.8811 | ✓ | ✓ |
Each folder contains model.safetensors and config.json (labels, image size,
input requirements, best macro AUROC).
The GAP variant only needs the CT volume. No segmentation or radiomics required.
pip install torch transformers safetensors nibabel scipy huggingface_hub
python inference.py ct.nii.gz --variant gap --top 10
Example output:
Disease Prob
──────────────────────────────────────────────────
Atherosclerosis 0.821 ████████████████
Renal Cyst 0.714 ██████████████
Hepatic Steatosis 0.431 ████████
...
These variants require organ segmentation masks (and radiomics features for fusion models). Use the full Janus preprocessing pipeline:
Source code: https://github.com/lavsendahal/janus (Repository is currently private — request access if needed.)
git clone https://github.com/lavsendahal/janus
cd janus && pip install -e .
python -m janus.inference --variant gated-fusion --ct ct.nii.gz
[B, 1, D, H, W] CT volume, HU clipped to [-1000, 1000] and normalised to [0, 1]Hepatomegaly, Splenomegaly, Cardiomegaly, Prostatomegaly, Hepatic Steatosis, Osteopenia, Gallstones, AAA, Aortic Valve Calcification, Coronary Calcification, Atherosclerosis, Thrombosis, Bowel Obstruction, Appendicitis, Hiatal Hernia, Submucosal Edema, Free Air, Biliary Ductal Dilation, Absent Gallbladder, Pancreatic Atrophy, Hydronephrosis, Renal Cyst, Renal Hypodensities, Pleural Effusion, Atelectasis, Ascites, Anasarca, Metastatic Disease, Lymphadenopathy, Fracture.
If you use this model, please cite the Janus paper (forthcoming).