pooyaeini/Aortic-valve-calcifications
GitHub: pooyaeini/Aortic-valve-calcifications
基于可解释机器学习的 TAVI 术后死亡率与并发症预测框架,整合术前多模态数据与主动脉瓣钙化拓扑特征,提供完全可复现的生存分析和分类管线。
Stars: 0 | Forks: 0
# TAVI 钙化拓扑机器学习再分析
## 概述
本仓库包含用于纽伦堡 TAVR 队列二次分析的可复现代码 (Pollari et al., *J Cardiovasc Comput Tomogr*, 2020; Mendeley Data: doi:10.17632/dkr773wtn3.1)。
### 研究目标
开发并进行内部验证可解释的机器学习模型,该模型将特定瓣叶的主动脉瓣和左心室流出道 (LVOT) 钙化拓扑与临床、超声心动图和 CT 预测因子相结合,用于:
- **主要终点:** TAVI 术后中期全因死亡率
- **次要终点:** 主要围手术期并发症的 VARC-2 复合终点
## 设计原则 (TRIPOD+AI / PROBAST+AI)
- **分析单元:** 患者
- **预测因子:** 仅限术前变量(临床、超声心动图、MDCT、钙化拓扑、外科风险评分、计划使用的器械/入路)
- **比较器:** EuroSCORE II
- **缺失数据:** 在每个交叉验证折叠中进行插补,以防止数据泄露
- **验证:** 采用带有折叠外预测的重复分层交叉验证
## 仓库结构
```
.
├── common.py
├── 01_survival_mortality.py
├── 02_binary_complication.py
├── 03_tables.py
├── run_all.py
├── build_manuscript.py
├── build_supplementary.py
├── build_response.py
├── requirements.txt
├── rawdata.xlsx
├── references.md
├── .gitignore
└── LICENSE
```
## 快速开始
### 安装依赖
```
pip install -r requirements.txt
```
主要软件包:
- scikit-survival
- lifelines
- shap
- python-docx
### 添加数据集
从 Mendeley Data 仓库下载 `rawdata.xlsx` 并将其放在项目根目录下。
### 运行完整 pipeline
```
python run_all.py
```
输出将写入:
- `figures/`
- `tables/`
- `results/`
### 生成手稿
```
python build_manuscript.py
```
这将生成 `Manuscript_TAVI_calcium_ML.docx`。
## 模型
### 生存模型
- EuroSCORE II (Cox PH)
- 惩罚 Cox (Coxnet)
- 梯度提升生存模型
- 随机生存森林
- 随机生存森林(不含钙化特征)
验证:
- 10 次重复 × 5 折分层交叉验证(50 折)
- C-index、时间依赖性 AUC、Integrated Brier Score、校准、决策曲线分析、Kaplan-Meier 分析
### 二分类并发症模型
- EuroSCORE II 逻辑回归
- L2 逻辑回归
- 随机森林
- 直方图梯度提升
验证:
- 20 次重复 × 5 折分层交叉验证
- AUROC、AUPRC、Brier 分数、校准、诊断性能、决策曲线分析、SHAP、亚组分析
## 分块执行
```
python 01_survival_mortality.py chunk 0 4
python 01_survival_mortality.py chunk 5 9
python 01_survival_mortality.py aggregate
```
## 可复现性
- 固定随机种子 (`RANDOM_STATE = 20240607`)
- 确定性交叉验证
- 保存折叠外预测
## 引用
请引用:
方法学参考文献:
- TRIPOD+AI (BMJ, 2024)
- PROBAST+AI (BMJ, 2025)
- 随机生存森林 (Ann Appl Stat, 2008)
- 决策曲线分析 (Med Decis Making, 2006)
## 许可证
MIT 许可证。原始数据集 (`rawdata.xlsx`) 仍受 Mendeley Data CC BY 4.0 许可证的约束。
标签:Apex, Python, TAVI手术预测, 医疗数据分析, 可解释性AI, 无后门, 机器学习, 生存分析, 逆向工具