VISTA3D Aorta Fine-tuned Checkpoint

Files

• model_finetune_aorta_seed42.pt

Training Summary

• Base model: nvidia/NV-Segment-CT
• Base checkpoint file: vista3d_pretrained_model/model.pt
• Local aorta label mapping: 1 -> 6
• Seed: 42
• Epochs: 20
• Best validation Dice: 0.8767 at epoch 7

Split

• Train: AD-005, AD-006, AD-007, AD-008, AD-011, AD-012, AD-014, AD-015, AD-019, AD-020
• Validation: AD-002, AD-013
• Held-out test: AD-001, AD-003, AD-004, AD-009, AD-016, AD-021

Held-out Test

• Mean Dice: 0.9123
• Median Dice: 0.9092
• Min Dice: 0.8274
• Max Dice: 0.9683