mlmed/torchxrayvision
TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.
What it solves
TorchXRayVision は、胸部X線データセットとディープラーニングモデルの取り扱いをシンプルにします。研究者が臨床課題のためにモデルをゼロから訓練する必要をなくし、メタデータ形式が異なることが多い複数の外部データセット間でモデルを評価するための統一インターフェースを提供します。
How it works
このライブラリは、さまざまな公開胸部X線データセットに対応した共通の前処理チェーンと標準化インターフェースを提供します。主に DenseNet121 と ResNet50 の事前学習済みモデルを含み、分類、表現学習、解剖学的セグメンテーションに利用でき、特徴抽出用の事前学習済みオートエンコーダーも提供します。
Who it’s for
大規模データセットの迅速な解析、Few‑Shot 学習のための特徴再利用、または異なるデータソース間でのモデル汎化性能の堅牢な評価が必要な医用画像研究者や AI 開発者向けです。
Highlights
- Pre-trained Models: NIH、CheXpert、MIMIC‑CXR など多様なコホートで訓練されたモデルにアクセスでき、ベースラインや特徴抽出器として利用可能。
- Unified Dataset Interface: 1 行のコードでデータセットを切り替え、マージやフィルタリングも簡単に行い、分布シフトの研究が可能。
- Anatomical Segmentation: 胸部X線の 14 種類の解剖構造をセグメントする事前学習済みモデルを提供。
- Pathology Masks: RSNA や SIIM などのデータセットに含まれる病理特化マスクをサポートし、より精密な解析が可能。
- Distribution Shift Tools:
CovariateDatasetなどのツールで共変量シフトやモデル汎化失敗をシミュレート・研究できます。