Upload 243 files

GRU.tar.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d2ff4a95881878dc102b2b031dd3eead19fc2e2a37e81e984f1a1a8bc981d8d7
+size 2468576

GRU/A1F50A2F50_A3F50/Dy_comp.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c70b5a7da868a4bb7ec522aaf04716bf08f4c643924400975abc7b85493d3dcc
+size 30620

GRU/A1F50A2F50_A3F50/Ex_vol.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e8bb8d8c87798e22c2a639fc0801a91907327ad781c40b0bc4fcd790a5e99af
+size 424972

GRU/A1F50A2F50_A3F50/P_mean.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28a158acc0b3ccd3c126cbbdfc03d53af643a3270449052b702abfa2fb9d896e
+size 59128

GRU/A1F50A2F50_A3F50/P_peak.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f07914281ab49625259e945c13722c995c428d0dbb89806a34aa1c92761c3a5
+size 179136

GRU/A1F50_A2F50/Dy_comp.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e295c9350e7eeb34ae3da4a7da7453fab3cc5df09dbbadb65393645305eb7df5
+size 8292

GRU/A1F50_A2F50/Ex_vol.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b74087d345133d816cdc8fb7e0026f15d7c77f66a2e90dcce98f9b15d4274cbd
+size 352432

GRU/A1F50_A2F50/P_mean.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d81e4a0fab8ce066717e769beb9bcef7bd0dd39066d0805375633c7d3b3180eb
+size 96628

GRU/A1F50_A2F50/P_peak.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fb38968f49034fbaaad1d5b31fe63abdef0a2c90b43e833a1f7dc207625b786a
+size 258572

GRU/A1F50_A3F50/Dy_comp.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56d19c90892c2cb9876f6d516527d92ef864dd90066c34d372ecf4240994398e
+size 374032

GRU/A1F50_A3F50/Ex_vol.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ebfb40081dcddbed61d4354feab4a2d0c5e8ff4068fe8b4bb9532ceccd5859f2
+size 385072

GRU/A1F50_A3F50/P_mean.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a21e1b10c2db12939c8a1819c0ac22ca73df7b531e89b82cc5ffae9f3a0d109
+size 236264

GRU/A1F50_A3F50/P_peak.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c33ff4950bfc8570fe6834a38a111269f105cac9dfb565c0351c7f855944640
+size 286652

README.md

@@ -0,0 +1,173 @@
+---
+license: cc-by-nc-nd-4.0
+language:
+- en
+tags: [digital-twin, ex vivo lung, time-series, forecasting, regression]
+
+---
+
+<h1 align="center">
+  <span style="font-size:32px">
+  Digital Twins of Ex Vivo Human Lungs
+  </span>
+</h1>
+
+<p align="center">
+  <a href="https://your-project-page-url">
+    <img alt="Project Page"
+         src="https://img.shields.io/badge/Project-Page-blue" />
+  </a>
+  <a href="https://github.com/Sage-Lab-ai/DT_Lung">
+    <img alt="GitHub Code"
+         src="https://img.shields.io/badge/GitHub-Code-black?logo=github" />
+  </a>
+  <a href="https://colab.research.google.com/drive/1IrzccQ09mP5amxQTZy07n4LgLSC2r1uj?usp=sharing">
+    <img alt="Colab: DT Lung Demo"
+         src="https://img.shields.io/badge/Colab-DT%20Lung%20Demo-orange?logo=google-colab" />
+  </a>
+  <a href="https://dt-lung.streamlit.app/">
+    <img alt="Streamlit Web App"
+         src="https://img.shields.io/badge/Streamlit-Web%20App-red?logo=streamlit" />
+  </a>
+  <a href="https://huggingface.co/SageLabUHN/DT_Lung">
+    <img alt="Hugging Face Model"
+         src="https://img.shields.io/badge/HuggingFace-Model-yellow?logo=huggingface" />
+  </a>
+  <a href="https://huggingface.co/datasets/SageLabUHN/DT_Lung_Demo_Data">
+    <img alt="Hugging Face Dataset"
+         src="https://img.shields.io/badge/HuggingFace-Dataset-yellow?logo=huggingface" />
+  </a>
+</p>
+
+## Purpose
+
+This is the official model repository that accompanies [Digital Twin of Ex Vivo Human Lungs Project](https://github.com/Sage-Lab-ai/DT_Lung). This repository contains *a full collection of multi-modal models for creating digital twins of ex vivo human lungs*.  
+
+**Authors**  
+<div align="center">
+<span style="font-size:16px">
+Xuanzi Zhou MHSc¹⁻³, Bo Wang PhD⁴⁻⁷, Yiyang Wei BCS¹⁻³, Serena Hacker MScAC<sup>1,2,4</sup>, Sumin Kim MSc<sup>1,2,4</sup>,  
+Thomas Borrillo MHSc<sup>1,2</sup>, Abby McCaig BSc¹⁻³, Haaniya Ahmed<sup>1,2</sup>, Youxue Ren MSc<sup>1,2</sup>,  
+Olivia Hough PhD<sup>1,2</sup>, Luca Orsini MD<sup>1,2,8</sup>, Bonnie T. Chao PhD<sup>1,2</sup>, Micheal McInnis MD⁹,  
+Marcelo Cypel MD<sup>1-3,10</sup>, Mingyao Liu MD<sup>1-3,10</sup>, Jonathan C. Yeung MD<sup>1-3,10</sup>,  
+Lorenzo Del Sorbo MD<sup>1,2,8</sup>, Shaf Keshavjee* MD<sup>1-3,10</sup>, and Andrew T. Sage* PhD<sup>1-3,10</sup>  
+</span>
+</div>
+
+<div align="center">
+<span style="font-size:14px">
+1. Latner Thoracic Research Laboratories, Toronto General Hospital Research Institute, University Health Network, Toronto, ON, Canada <br> 
+2. Toronto Lung Transplant Program, Ajmera Transplant Centre, University Health Network, Toronto, ON, Canada  <br>
+3. Institute of Medical Science, University of Toronto, Toronto, ON, Canada  <br>
+4. Department of Computer Science, University of Toronto, Toronto, ON, Canada  <br>
+5. Peter Munk Cardiac Centre, University Health Network, Toronto, ON, Canada <br> 
+6. Department of Laboratory Medicine and Pathobiology, University of Toronto, Toronto, ON, Canada  <br>
+7. Vector Institute, Toronto, ON, Canada  <br>
+8. Interdepartmental Division of Critical Care Medicine, Medical and Surgical Intensive Care Unit, University Health Network, Toronto, ON, Canada  <br>
+9. Department of Medical Imaging, Temerty Faculty of Medicine, University of Toronto, University Health Network, Toronto, ON, Canada  <br>
+10. Department of Surgery, Temerty Faculty of Medicine, University of Toronto, Toronto, ON, Canada <br> 
+</span>
+</div>
+
+<div align="center">
+<small>*Co-Senior authors</small>
+</div>
+
+
+---
+
+## Highlights
+- ✅ Trained on the world's largest ex vivo lung function dataset  
+- ⚡ Fast inference: Creating a digital lung within minutes
+- 📊 Predicts over 75 multi-modal lung functional parameters including lung physiology, biochemistry, images, protein and transcriptomic markers.
+- 🔄 Supports both static (single‑timepoint) and dynamic (continuous recalibration) forecasting  
+
+---
+
+## Available Models
+The DT model is built using two machine learning model architectures: gated recurrent unit (GRU) and XGBoost (XGB).
+
+1. **GRU‑based Time‑Series Models**
+
+  - `DT_Lung/GRU`
+    - `A1F50_A2F50` – Collection of GRU models for static digital lung forecasting (Forecast 2<sup>nd</sup> hour lung function using 1<sup>st</sup> hour baseline data)
+    - `A1F50_A3F50` – Collection of GRU models for static digital lung forecasting (Forecast 3<sup>rd</sup> hour lung function using 1<sup>st</sup> hour baseline data)
+    - `A1F50A2F50_A3F50` – Collection of GRU models for dynamic digital lung forecasting (Forecast 3<sup>rd</sup> hour lung function using 1<sup>st</sup> and 2<sup>nd</sup> hour observed data)
+
+     Legend: `A = assessment period`, `F = first breaths`, `numbers = the number of breaths included` <br>
+     *Note: everything before _ is included as input to the model, and everything after _ is forecasted by the model.* <br>
+     Each folder contains the best-performing GRU models resulting from our hyperparameter tuning for the specified digital lung setup.
+
+2. **XGBoost Tabular Regressors** 
+
+   Each folder below contains the best-performing XGBoost models resulting from our hyperparameter tuning for the specified lung function parameters for each digital lung setup.
+
+  - `DT_Lung/XGB` – Collection of XGBoost models for multiple data modalities
+    - `Hourly` – XGBoost models for predicting hourly lung functional parameters (e.g., oxygenation level, compliance, pH, etc.)
+      - `H1_to_H2`: Models for static digital lung forecasting (Forecast 2nd hour lung function using 1st hour baseline data)
+      - `H1_to_H3`: Models for static digital lung forecasting (Forecast 3rd hour lung function using 1st hour baseline data)
+      - `H1_H2_to_H3`: Models for dynamic digital lung forecasting (Forecast 3rd hour lung function using 1st and 2nd hour observed data)
+    - `Protein` – XGBoost models for predicting protein markers (e.g., interleukin-8, interleukin-6, etc.)
+      - `H1_to_H2`: Models for static digital lung forecasting (Forecast 2nd hour lung function using 1st hour baseline data)
+      - `H1_to_H3`: Models for static digital lung forecasting (Forecast 3rd hour lung function using 1st hour baseline data)
+      - `H1_predH2_to_H3`: Models for static digital lung forecasting (Forecast 3rd hour lung function using 1st hour baseline data and predicted 2nd hour data)
+      - `H1_H2_to_H3`: Models for dynamic digital lung forecasting (Forecast 3rd hour lung function using 1st and 2nd hour obsereved data)
+    - `Transcriptomics` – XGBoost models for predicting transcriptomic pathways (e.g., TGF-β, apoptosis, etc.)
+      - `static_forecasting`: Models for static digital lung forecasting of gene enrichment scores (Forecast post-perfusion gene enrichment scores using baseline data)
+      - `dynamic_forecasting`: Models for dynamic digital lung forecasting of gene enrichment scores (Forecast post-perfusion gene enrichment scores using hourly observed data)
+      
+
+  - `DT_Lung/XGB_PC` – Collection of XGBoost models for lung x-ray images
+    - `models_static` – XGBoost models to create static digital lung image features (Forecast 3<sup>rd</sup> hour lung X-ray derived features using 1<sup>st</sup> hour baseline data)
+    - `models_dynamic` – XGBoost models to create dynamic digital lung image features (Forecast 3<sup>rd</sup> hour lung X-ray derived features using 1<sup>st</sup> and 2<sup>nd</sup> hour observed data)
+
+3. Folders with the `.tar.gz` extension are provided to make model download and distribution easier. Their contents are identical to those in the corresponding folders without the `.tar.gz` extension. 
+
+---
+
+## Downloading Models
+
+### Option 1: Download individual models
+
+```
+from huggingface_hub import hf_hub_download
+
+gru_model_dir = hf_hub_download(
+    repo_id="SageLabUHN/DT_Lung",
+    filename="GRU/A1F50_A2F50/Dy_comp.pt", # example path, please modify accordingly
+    local_dir="DT_Lung/models"  # example path, specify your local directory
+)
+
+print(f"Model downloaded to: {gru_model_dir}")
+```
+
+### Option 2: Download all models to a specific folder
+
+```
+from huggingface_hub import snapshot_download
+
+model_dir = snapshot_download(
+    repo_id="SageLabUHN/DT_Lung",
+    local_dir="DT_Lung/Model", # make sure the Model folder is in the project root dir
+)
+
+print(f"Models downloaded to: {model_dir}")
+```
+
+Alternatively, you can manually download the models from the [Hugging Face repository page](https://huggingface.co/SageLabUHN/DT_Lung/tree/main).
+
+## Intended Use
+
+**Intended for**  
+- Research and education use 
+
+**Not for**  
+- Clinical decisions without further validation  
+- Commercial deployment (non‑commercial only) 
+
+---
+
+## License
+
+This repository and all model weights are released under the **Creative Commons Attribution‑NonCommercial‑ShareAlike 4.0 International (CC BY‑NC‑SA 4.0)** license.  
+Commercial use is **prohibited**.

XGB.tar.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a85f4060d9bed511fc91e64b02cce8fc767533a484cb2b76b96f8d563662df75
+size 5626781

XGB/Hourly/H1_H2_to_H3/90_Calc Delta PCO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8f2fb6616fadc6034ad1012281d411eefcd8e434f6612d7183eda4adac99cd2
+size 349191

XGB/Hourly/H1_H2_to_H3/90_Cdyn.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37573ae28f0ee97c330b8e33ed1a068fb4582436ca588207b5e1ce693002252c
+size 62054

XGB/Hourly/H1_H2_to_H3/90_Cstat.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40baaa55c15687620f0cfb369e26ffe0c6574a508cb50750074a1a20f9fbd5e0
+size 63479

XGB/Hourly/H1_H2_to_H3/90_Delta PO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7940d8be3c88aab01cbf904f0a4d5dee91609db3572896970485c22da1671f72
+size 60006

XGB/Hourly/H1_H2_to_H3/90_LA BE.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f11cc3a979968c562b229bc2fd5e21d7f863f92cb51fbf54ab85d11d73b086a6
+size 62671

XGB/Hourly/H1_H2_to_H3/90_LA CL.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa44fa19ef4429de0cb9622e9dc98074b7b16dd9a797991e19993b9cac6e3e45
+size 661619

XGB/Hourly/H1_H2_to_H3/90_LA Ca++.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db80933aaeb7c1ac3ad4640d3460195f97907ce5a886dca460b3c78921110b0f
+size 92737

XGB/Hourly/H1_H2_to_H3/90_LA Glu.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df69742843b93b3a57b047a35c021d5c54839460e4398b14cf42914f55d5cba8
+size 111747

XGB/Hourly/H1_H2_to_H3/90_LA HCO3.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a110b8330dce55585a2516738557abeff026feb3b10275d5ed9bcf50177b24ad
+size 237387

XGB/Hourly/H1_H2_to_H3/90_LA K+.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cab18145732466bca83aac2d63c5dbf7654db7bfcf25fed581888ae28e2bbe2
+size 113577

XGB/Hourly/H1_H2_to_H3/90_LA Lact.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71ef0592ba7009472ccb8f42d57afc93c4581e72b1cc9c68f3025af284c8e07a
+size 62677

XGB/Hourly/H1_H2_to_H3/90_LA Na+.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8719e4446f86fe7e0286315e3296c9e14ea3462dff1fbf12461fa0b2dfeb719d
+size 335040

XGB/Hourly/H1_H2_to_H3/90_LA PCO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ca8c10cb40c7e1a4e94e186fa401b914ff4f52d0f7c3d960b726791e9af81ab
+size 115218

XGB/Hourly/H1_H2_to_H3/90_LA PO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4210f7b8f821bf29e771a41f3daa055532a1e78646d1ceaa2e14619ac1563dc
+size 63216

XGB/Hourly/H1_H2_to_H3/90_LA pH.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ca4307e7070874223f8b459851570993e6c1131629fd5809f2c7865d0d88380
+size 125715

XGB/Hourly/H1_H2_to_H3/90_LAP.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f103454b75a9570699eeda14cce40be9324360a5a2e7d617a9afe98b033b4d10
+size 27222

XGB/Hourly/H1_H2_to_H3/90_PA PCO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5aef4eb040d7e7a9e448ddbd33f6720915ec897ff66eea686a7f263daecbc6a4
+size 62331

XGB/Hourly/H1_H2_to_H3/90_PA PO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b2a7b821107d52ac43ffd7f3c7ff9608683029edd010b016f9361dbfd8d0d804
+size 61719

XGB/Hourly/H1_H2_to_H3/90_PAP.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8b0ec5c0fd8269d38d8ea0dde7d4452fb77005f7f5bda0e4623b32f53e9ceb7
+size 329976

XGB/Hourly/H1_H2_to_H3/90_STEEN lost.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c64e5842a2a997bea6ccb33e18af50a84a597bef974de6df1c41df5c334b0298
+size 334386

XGB/Hourly/H1_H2_to_H3/90_pMean.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:352a5d2a49bb5b16d5119f8c2341ba3014ee7f063cdee108e4d01e1b1102a42d
+size 54954

XGB/Hourly/H1_H2_to_H3/90_pPeak.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9925d9f9cee32499f36356d791d6ab81c2da0d677d867b87de05ecc7a89bd25f
+size 61715

XGB/Hourly/H1_H2_to_H3/90_pPlat.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d5626b3cfd445277da6f535f1ff9872adb4f970268f2dd5b57245f349ed94e77
+size 61930

XGB/Hourly/H1_to_H2/80_Calc Delta PCO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15b124e72f91b7fd535fab2bd83e488e260b097569069bc593bb63644dc18c70
+size 93287

XGB/Hourly/H1_to_H2/80_Cdyn.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3bde3cb16038cfe5897f763d2b4481834d27072c75d154939ff8db00c47d3b67
+size 62397

XGB/Hourly/H1_to_H2/80_Cstat.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86bf51afc35ba5d45a766a842871c1def7d7e4ba60d1fc1e3563799ee8c85a23
+size 61715

XGB/Hourly/H1_to_H2/80_Delta PO2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab20d6a238d3d9901f1d90038803c435b4da822253323ef005e91617ac9447b2
+size 62263

XGB/Hourly/H1_to_H2/80_LA BE.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce828dba0505595b2d58c623fa6e91f58ccb759a10f9821793d8c6adc1031000
+size 62269

XGB/Hourly/H1_to_H2/80_LA CL.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c4d288b386143d983a5b007db8aa81b48dc8c849ee31c99c158b30633f9d1e65
+size 62125

XGB/Hourly/H1_to_H2/80_LA Ca++.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e311ce591ff765a6732deb70fdcdbca1d4ddf95f1551cedacb652ae37cf3f4b7
+size 93076

XGB/Hourly/H1_to_H2/80_LA Glu.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e17de32bb21bdec51de1bec99d0acf5d2c5e2c89adb8ae777c1fb4d7a4b01f9
+size 43949

XGB/Hourly/H1_to_H2/80_LA HCO3.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de86a877355b4186a0dad442aa8bf06fa81537c7862d063ec6fa4fc1a5a5ca66
+size 106280

XGB/Hourly/H1_to_H2/80_LA K+.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e32a266cd7c0d3e2d2819d2f980510ecaf74d0ade56ecfa08777e2141ea7ed9c
+size 93349

XGB/Hourly/H1_to_H2/80_LA Lact.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab0a948c0f016b75aec50c79d6134e699c92397410992bccb501934f28ec675b
+size 276799

XGB/Hourly/H1_to_H2/80_LA Na+.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f87fc5be96bc272a7bc235068349e4491116a63bc904d5d0635a993556dd995b
+size 346995