Enhance model card with metadata, abstract, and GitHub link

README.md

@@ -2,24 +2,26 @@
 language: en
 license: cc-by-nc-sa-4.0
 tags:
-  - survival-analysis
-  - multiple-instance-learning
-  - optimal-transport
-  - medical-imaging
-  - deep-learning
-  - pytorch
+- survival-analysis
+- multiple-instance-learning
+- optimal-transport
+- medical-imaging
+- deep-learning
+- pytorch
+pipeline_tag: image-classification
+library_name: pytorch
 model-index:
-  - name: OTSurv
-    results:
-      - task:
-          type: survival-analysis
-          name: Survival Prediction
-        dataset:
-          type: TCGA
-          name: TCGA (BLCA, BRCA, LUAD, STAD, COADREAD, KIRC)
-        metrics:
-          - type: c-index
-            value: 0.646
+- name: OTSurv
+  results:
+  - task:
+      type: survival-analysis
+      name: Survival Prediction
+    dataset:
+      name: TCGA (BLCA, BRCA, LUAD, STAD, COADREAD, KIRC)
+      type: TCGA
+    metrics:
+    - type: c-index
+      value: 0.646
 ---
 
 <div align="center">
@@ -55,6 +57,9 @@ model-index:
   <a href="https://huggingface.co/datasets/Y-Research-Group/OTSurv_Dataset">
     <img src="https://img.shields.io/badge/Hugging%20Face-Dataset-green?style=flat-square&logo=huggingface" alt="Hugging Face Dataset">
   </a>
+  <a href="https://github.com/Y-Research-SBU/OTSurv">
+    <img src="https://img.shields.io/badge/GitHub-Code-181717?style=flat-square&logo=github" alt="GitHub Code">
+  </a>
   <a href="#">
     <img src="https://img.shields.io/badge/PyTorch-2.3-EE4C2C?style=flat-square&logo=pytorch" alt="PyTorch 2.3">
   </a>
@@ -62,6 +67,9 @@ model-index:
 
 </div>
 
+## Abstract
+
+Survival prediction using whole slide images (WSIs) can be formulated as a multiple instance learning (MIL) problem. However, existing MIL methods often fail to explicitly capture pathological heterogeneity within WSIs, both globally -- through long-tailed morphological distributions, and locally through -- tile-level prediction uncertainty. Optimal transport (OT) provides a principled way of modeling such heterogeneity by incorporating marginal distribution constraints. Building on this insight, we propose OTSurv, a novel MIL framework from an optimal transport perspective. Specifically, OTSurv formulates survival predictions as a heterogeneity-aware OT problem with two constraints: (1) global long-tail constraint that models prior morphological distributions to avert both mode collapse and excessive uniformity by regulating transport mass allocation, and (2) local uncertainty-aware constraint that prioritizes high-confidence patches while suppressing noise by progressively raising the total transport mass. We then recast the initial OT problem, augmented by these constraints, into an unbalanced OT formulation that can be solved with an efficient, hardware-friendly matrix scaling algorithm. Empirically, OTSurv sets new state-of-the-art results across six popular benchmarks, achieving an absolute 3.6% improvement in average C-index. In addition, OTSurv achieves statistical significance in log-rank tests and offers high interpretability, making it a powerful tool for survival prediction in digital pathology. Our codes are available at this https URL .
 
 ## 🧠 DL;TR
 
@@ -184,6 +192,12 @@ cd src
 python analysis/plot_survival_curv.py
 ```
 
+The survival curve for TCGA-BLCA looks like this:
+<div align="center">
+  <img src="result/visulization/BLCA_km.png" alt="TCGA-BLCA Survival Curve" width="500"/>
+</div>
+
+<br>
 
 ## 📊 Performance Results