English注册

Adilbai/bone-age-resnet-80m

来源：🤗image-segmentationmit61M 参数243 MBsafetensors✓ 3 个校验和497 天前更新
或提交种子

🦴 Bone Age Regression Model

🚀 Quick Start

📋 Model Overview

🎯 Predicts bone age from hand X-rays with ~5 month accuracy
This CNN-based model uses ResNet152 architecture to estimate pediatric bone age from hand radiographs, achieving an MSE of ~25 (equivalent to ±5 month prediction range).

🏥 Clinical Impact

  • Accuracy: MSE ~25 months² (±5 month typical error range)
  • Speed: Real-time inference (<1 second per image)
  • Applications: Pediatric growth assessment, endocrine disorder screening
  • Support: Assists radiologists in bone age evaluation

🧠 Architecture Components

  • 🏗️ Base Model: ResNet152 (80M+ parameters)
  • 🔄 Pre-training: ImageNet initialization
  • 🎯 Task Head: Custom regression layers
  • 👥 Multi-modal: Image + gender fusion
  • 📐 Input Size: 256×256 RGB images

📊 Performance Metrics

Metric Value Interpretation
MSE ~25 months² ±5 month typical error
Training Loss 1567.98 → 25.26 98.4% improvement
Convergence 9 epochs Stable training
Speed 1.69 it/s Real-time capable

🎯 Intended Use Cases

✅ Recommended Uses ❌ Not Recommended
🏥 Clinical decision support 🚫 Standalone diagnosis
📚 Medical education 🚫 Adult bone age
🔬 Research applications 🚫 Non-hand X-rays
👨‍⚕️ Radiologist assistance 🚫 Emergency decisions

📊 Training Performance

📈 Training Progress

Epoch Loss Improvement Status
1 1567.98 - 🔴 Starting
2 178.89 -88.6% 🟡 Learning
5 63.82 -95.9% 🟠 Converging
9 24.15 -98.5% 🟢 Best
10 25.26 -98.4% 🔵 Final

📋 Training Configuration

  • 📦 Dataset: RSNA Bone Age (12,500 images)
  • ⏱️ Duration: ~1.5 hours (10 epochs)
  • 🎯 Optimization: SGD/Adam (details in code)
  • 📊 Batch Size: ~32 (395 batches/epoch)
  • 🔄 Best Checkpoint: Epoch 9 (MSE: 24.15)

🚀 Usage Examples

🐍 Python - PyTorch

# 📦 Installation
pip install torch torchvision pillow

# 🔮 Inference
from PIL import Image
import torch
from finetune_resnet_bone_age import BoneAgeResNet, transforms

# 📥 Load model
model = BoneAgeResNet()
model.load_state_dict(torch.load('resnet_bone_age_80m.pt'))
model.eval()

# 🖼️ Prepare inputs
image = Image.open('hand_xray.png').convert('RGB')
img_tensor = transforms(image).unsqueeze(0)
gender = torch.tensor([0.0])  # 0=male, 1=female

# 🎯 Predict
with torch.no_grad():
    predicted_age = model(img_tensor, gender)
    print(f"🦴 Predicted bone age: {predicted_age.item():.1f} ± 5 months")

⚡ ONNX Runtime

import onnxruntime as ort
import numpy as np

# 🔧 Load ONNX model
session = ort.InferenceSession('resnet_bone_age_80m.onnx')

# 🎯 Run inference
outputs = session.run(None, {
    "image": img_array,
    "gender": np.array([[0.0]])  # 0=male, 1=female
})

age_months = outputs[0][0]
print(f"🦴 Bone age: {age_months:.1f} months ({age_months/12:.1f} years)")

📚 Related Work & Background

🔬 Scientific Foundation

Bone age assessment is a critical clinical tool in pediatric medicine, traditionally performed using the Greulich-Pyle or Tanner-Whitehouse methods. Deep learning approaches have shown promising results in automating this process.

📖 Key Publications

  • Larson et al. (2018): "Performance of a Deep-Learning Neural Network Model in Assessing Skeletal Maturity on Pediatric Hand Radiographs" - Radiology
  • Iglovikov et al. (2018): "Paediatric Bone Age Assessment Using Deep Convolutional Neural Networks" - MICCAI
  • Liu et al. (2019): "Bone Age Assessment Based on Deep Convolution Features" - Frontiers in Neuroscience

🧠 CNN Architecture Evolution

  • Traditional CNNs: AlexNet, VGG → Limited medical imaging performance
  • ResNet Revolution: Skip connections → Better gradient flow, deeper networks
  • Medical Adaptations: Transfer learning + domain-specific fine-tuning
  • Multi-modal Integration: Image + metadata fusion for improved accuracy

🔄 Comparison with Other Approaches

Method Architecture MSE Year
Greulich-Pyle (Manual) Human Expert ~20-30 1959
This Model ResNet152 ~25 2024
Iglovikov et al. VGG-16 ~30-35 2018
Larson et al. CNN Ensemble ~15-20 2018

⚠️ Important Limitations

🎯 Accuracy Interpretation

MSE ≈ 25 months² means typical errors of ±5 months

🏥 Clinical Considerations

  • 📋 FDA Status: Not FDA approved - research use only
  • 👨‍⚕️ Professional Oversight: Requires medical supervision
  • 🎯 Population: Validated on RSNA dataset demographics
  • ⚖️ Bias: May vary across different ethnic groups

🔧 Technical Limitations

  • 📸 Image Quality: Requires clear, properly positioned hand X-rays
  • 👶 Age Range: Optimized for pediatric patients (0-18 years)
  • 💾 Memory: ~1GB RAM required for inference
  • ⚡ Hardware: GPU recommended for real-time performance

🚀 Deployment Options

🔧 Quick Deploy

🐳 Docker Deployment

FROM pytorch/pytorch:latest
COPY requirements.txt .
RUN pip install -r requirements.txt
COPY . /app
WORKDIR /app
EXPOSE 8000
CMD ["python", "app.py"]

☁️ Cloud Integration

  • Hugging Face Inference API: Serverless deployment
  • AWS Lambda: Cost-effective inference
  • Google Cloud Run: Scalable container deployment
  • Azure Container Instances: Enterprise integration

📊 Model Card Information

📈 Performance Summary

  • 🎯 Task: Bone age regression from hand X-rays
  • 📊 Metric: Mean Squared Error (MSE)
  • 🏆 Score: ~25 months² (±5 month error range)
  • ⚡ Speed: Real-time inference capability
  • 💾 Size: ~320MB (PyTorch), ONNX compatible

🔬 Training Details

  • 📦 Dataset: RSNA Bone Age (12,500 images)
  • 🏗️ Architecture: ResNet152 + custom regression head
  • ⚙️ Parameters: 80+ million
  • 📊 Epochs: 10 (best at epoch 9)
  • 🔄 Convergence: 98.4% loss reduction

📋 Citation

@model{adilbai2024bone_age_resnet,
  title={Bone Age Regression Model (ResNet152, 80M+ params)},
  author={Adilbai},
  year={2024},
  url={https://huggingface.co/Adilbai/bone-age-resnet-80m},
  note={MSE ~25 months², ±5 month typical error}
}

🤝 Community & Support

💡 Contributing

We welcome contributions! Please see our contribution guidelines for details.

📞 Contact

⚠️ Medical Disclaimer: This model is for research and educational purposes only. Not intended for clinical diagnosis without proper medical supervision and validation.