🚀 SPIDER-Thorax模型
SPIDER-thorax模型是一個用於圖像分類的深度學習模型,專門針對胸部的切片級病理分類進行訓練。它是SPIDER數據集計劃的一部分,該計劃提供了一個大型、高質量、多器官的病理數據集,並帶有專家標註的標籤。
🚀 快速開始
模型適用於 1120×1120 的切片。使用以下代碼片段來加載和使用模型:
from transformers import AutoModel, AutoProcessor
from PIL import Image
model = AutoModel.from_pretrained("histai/SPIDER-thorax-model", trust_remote_code=True)
processor = AutoProcessor.from_pretrained("histai/SPIDER-thorax-model", trust_remote_code=True)
image = Image.open("path_to_image.png")
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
print(outputs.predicted_class_names)
✨ 主要特性
- 專為胸部切片級病理分類設計。
- 基於SPIDER數據集進行訓練,數據質量高且有專家標註。
📦 安裝指南
文檔未提及具體安裝步驟,可參考 transformers
庫的安裝方式。
📚 詳細文檔
模型描述
SPIDER-thorax模型是一個用於切片級病理分類的深度學習模型,特別針對胸部。它是SPIDER數據集計劃的一部分,該計劃提供了一個大型、高質量、多器官的病理數據集,並帶有專家標註的標籤。
如果您想支持、贊助或獲取SPIDER數據和模型的商業許可,請通過 models@hist.ai 與我們聯繫。
模型來源
訓練數據
該模型在 SPIDER-thorax 數據集上進行訓練,這是SPIDER數據集的一個子集。該數據集包含:
類別 |
中心切片數量 |
肺泡 |
6652 |
支氣管軟骨 |
5685 |
支氣管腺體 |
4412 |
慢性炎症 + 纖維化 |
6070 |
碎屑 |
5146 |
纖維化 |
6494 |
出血 |
5247 |
淋巴結 |
6088 |
色素 |
5177 |
胸膜 |
4560 |
非小細胞腫瘤 |
6445 |
小細胞腫瘤 |
5061 |
軟組織腫瘤 |
5894 |
血管 |
5376 |
總數統計:
- 78,307 箇中心切片
- 599,459 個總切片(包括上下文切片)
- 411 個用於標註的總切片
結果
器官 |
準確率 |
精確率 |
F1分數 |
胸部 |
0.962 |
0.958 |
0.960 |
📄 許可證
該模型遵循 CC BY-NC 4.0 許可協議,僅用於 研究用途。
🔗 引用
如果您使用此模型,請引用以下內容:
@misc{nechaev2025spidercomprehensivemultiorgansupervised,
title={SPIDER: A Comprehensive Multi-Organ Supervised Pathology Dataset and Baseline Models},
author={Dmitry Nechaev and Alexey Pchelnikov and Ekaterina Ivanova},
year={2025},
eprint={2503.02876},
archivePrefix={arXiv},
primaryClass={eess.IV},
url={https://arxiv.org/abs/2503.02876},
}
ℹ️ 更多信息
要探索其他模型和SPIDER數據集,您可以訪問項目的 Hugging Face HistAI頁面 或 GitHub倉庫。
📞 聯繫方式
- 作者:Dmitry Nechaev, Alexey Pchelnikov, Ekaterina Ivanova
- 郵箱:dmitry@hist.ai, alex@hist.ai, kate@hist.ai