← 返回项目← All projects

CausalResearch

治疗方案因果推荐 · WIP · 2026

Causal Treatment Recommendation · 2026

Some heart patients have three surgical options — but each patient only ever gets one. This research builds an AI that asks: "what if we had chosen differently?"

一些心脏病患者面临三种手术方案——但每个病人永远只会经历其中一种。这项研究造了一个会问「当初如果选另一种会怎样?」的 AI。

方案 A方案 B方案 C 三条路,只能走一条
摘要 / ABSTRACT ABSTRACT 每位病人留下三类痕迹——血管影像、放射报告与化验表格;三路编码器(ResNet / BERT / TabPFN)经门控融合为患者表征,Multi-DragonNet 对三种手术方案各自预测「假想结局」,以排序损失训练推荐器。小队列(835 训练 / 209 测试)上,治疗有效时与医生约 93% 一致;识别无效治疗仍是开放难题。 Every patient leaves three kinds of traces — angiography images, radiology reports and lab tables. Three encoders (ResNet / BERT / TabPFN) gate-fuse them into one patient representation, from which a Multi-DragonNet predicts the counterfactual outcome of each of three surgical options, trained with a ranking loss. On a small cohort (835 train / 209 test) the model agrees with doctors ~93% of the time when treatment helped; recognizing failed treatments remains open.

The medical setting, in plain words

用大白话讲讲背景

For elderly heart patients there is a common minimally invasive surgery — in this cohort it comes in three variants. Doctors choose one — and that's the only outcome anyone ever observes.

一些老年心脏病患者需要接受一种常见的微创手术——在这个队列里,这种手术有三种做法,医生只能选一种——而世上能观察到的,也只有这一种的结果。

The road not taken is invisible. Yet the whole point of a good recommendation is to know what the other roads would have done. That's the counterfactual problem — the heart of causal inference.

没选的那条路,永远看不见。但「推荐」的意义恰恰在于知道别的路会怎样——这就是反事实问题,因果推断的核心难题。

The trick: three clues about each patient

破局思路:每位病人的三条线索

Each patient leaves three kinds of traces: blood-vessel scans (images), the radiologist's written report (text), and lab-sheet numbers (tables). The model reads all three and compresses them into one "digital portrait" of the patient.

每位病人留下三类痕迹:血管影像(图像)、放射科医生写的报告(文字)、化验单(表格)。模型把三者一起读入,压缩成这位病人的一个「数字画像」。

CausalResearch model architecture
Fig. 1 模型总览:三模态(影像 / 化验 / 报告)分别编码后门控融合,Multi-DragonNet 输出各治疗的反事实结局 Ŷ。 Fig. 1 Model overview: three modalities encoded separately, gate-fused into a patient representation; Multi-DragonNet outputs the counterfactual outcome Ŷ of each treatment.
MRA 血管影像 放射报告(文字) 年龄 79EF 55%肌酐 1.2 化验表格 256 维「数字画像」 三类信息融合成一个向量 缺失某一类?用可学习的「缺失占位符」补上,并做标记

Then: predict all three "what ifs"

然后:预测三种「如果」

From the portrait, the model outputs a predicted outcome for each of the three options — including the two that never happened — and recommends the best. A special "ranking loss" trains it: when the real treatment worked, the model must rank it above the alternatives; when it failed, it must find something better.

模型从画像出发,对三种方案各自输出一个预测结果——包括从未发生的那两种——然后推荐最好的。一种特殊的「排序损失」负责训练:真实方案有效时,模型必须把它排在别的方案前面;真实方案无效时,模型必须找到更好的替代。

数字画像 DragonNet 倾向性校正 减少选择偏差 TransTEE Transformer 版对照 A 预后 62 B ★ 预后 88 C 预后 71 推荐 B 给医生参考 对三种方案各估一个"假想结局",从未发生的两种也在内

Where it stands today

目前的进展

1,044patients in experiments实验病例数
3treatment options scored评估的方案数
93%agreement when treatment worked治疗有效时推荐一致率
🚧spotting failed treatments: still weak识别无效治疗:仍是短板

On a small cohort (835 training / 209 test), when the real treatment did help, the model agrees with doctors ~93% of the time. Recognizing that a treatment failed — and suggesting something better — remains the open challenge. Research continues.

在小队列上(835 训练 / 209 测试):当真实治疗确实有效时,模型与医生的选择约 93% 一致;而识别「治疗无效、应换方案」仍是待解难题。研究仍在继续。