DiscoCATEHCGM working paper
Manuscript v0 · Theory draft DISC-CATE-WP-2026-07-17-V1

Heterogeneous treatment effects · localization · epistemic uncertainty

先决定谁能为这个 unit 发言,
再估计它附近的因果效应。

当 focal unit 不能被一个 point representation 充分描述时,能否用它对 latent response types 的完整 belief,重新定义哪些 observed samples 应当为它提供因果证据?

Causal Forest 在 observed covariate geometry 中学习自适应邻域。DiscoCATE 先从 pretreatment evidence 得到 Unit-Abducted belief \(Q_i\),再把 belief proximity 拉回 observed donor rows,最后局部平均 cross-fitted causal scores。

已经完成fixed-\(Q\) localized target 的识别论证
正在建立learned-\(Q\) stability 与 matched evaluation
尚未声称benchmark superiority、inference theorem 或 latent recovery

One-minute thesis

这篇论文真正改变的是 neighborhood 的定义

每个 nonparametric CATE estimator 都必须回答:哪些 observed units 可以为 focal query 提供信息,以及各自应有多大权重。DiscoCATE 的新对象不是另一个 outcome model,而是一张 probability-measure geometry。

一句话主张:HTE localization 不一定只能建立在 observed covariate geometry 上, 也可以建立在由 unit beliefs 诱导的 response-relevant geometry 上。

但 \(Q_X\) 仍是 \(X\) 的函数。它不会创造新的 individual-level information;可能的收益只能来自 mechanism-informed inductive bias、对 multimodal uncertainty 的保留,以及更合适的有限样本 borrowing。

Same computational grammar · different sample spaces

Causal Forest 与 DiscoCATE 共享什么,又绝不能混淆什么?

Observed-unit space01

Causal Forest / GRF

森林通过 learned partitions 与 co-leaf relations,决定哪些 observed training rows 应为 \(x_i\) 提供局部因果证据。

12345
\(\alpha_j(x)\longrightarrow\) localized causal moment
Belief-induced donor space02

DiscoCATE

先比较关于 fixed focal tokens 的 complete beliefs,再把这种 proximity 转换为 observed donor-row weights。

abcde
\(Q_i,Q_j\longrightarrow\alpha_{ij}\longrightarrow\sum_j\alpha_{ij}\Gamma_j\)
两个权重不能混为一谈。

\(Q_i\) 在 latent candidate-type space 上表达对同一个固定 token 的 epistemic uncertainty; \(\alpha_{ij}\) 在 observed sample space 上决定 donor row \(j\) 的贡献。DiscoCATE 从前者诱导后者, 并不是直接用 \(Q_i\) 替换 forest weights。

Two similarity constructions

两个 units 在这里怎样才算“相似”?

论文保留了两条 pretreatment-only 路线,但当前只把 belief distance 作为主算法;同时明确划出 outcome-conditioned likelihood 的边界。

A
当前默认算法

Belief-to-belief distance

\[\alpha_{ij}^{(h)}\propto K_h\{d(Q_i,Q_j)\}.\]

直接比较两个完整 probability measures。Hellinger 或 Jensen–Shannon 可以作为 bounded 主方案; symmetrized KL 是有价值的 sensitivity analysis,但 support mismatch 时可能发散。

  • 保留 multimodality 与 dispersion;
  • posterior mean、hard class、raw-\(X\) distance 是 baselines;
  • 距离与 bandwidth 都必须报告 sensitivity。
B
已定义的候选扩展

Focal-belief evidence compatibility

\[s^X_{i\to j}=\int p_\psi(X_j\mid u)\,Q_i(du).\]

它问:“如果 focal belief 中的 candidate types 是合理的,donor \(j\) 的 pretreatment evidence 是否 compatible?”这是方向性的 similarity,需要自己的 normalization、support 与 stability analysis。

  • 只读 pretreatment evidence;
  • 依赖一个冻结的 generative evidence model;
  • 尚未获得与路线 A 同等完整的 estimator theory。
不进入当前 ex-ante CATE estimator

Outcome-conditioned posterior-predictive similarity

\[s^{Y,T}_{i\to j}=\int p_\theta(Y_j,T_j\mid X_j,u)\,Q_i(du).\]

如果 realized \(Y_j\) 或 \(T_j\) 决定 donor weight,而同一 row 又贡献 causal score,就发生 outcome-dependent selection。Cross-fitting likelihood model 并不会自动恢复原 CATE estimand。 这种分数可以对应 ex-post matching problem,但不属于当前 DiscoCATE-DR。

DiscoCATE-DR

Belief geometry 决定向谁借信息,AIPW 负责 causal signal

新 geometry 不被要求兼任 propensity score,也不能自动修复 hidden confounding。

01Outer-fold abductor

用 external data 或 outer folds 训练 \(\widehat{\mathcal Q}\)。

02Pretreatment belief

只从 \(X_i\) 生成冻结的 \(\widehat Q_i\)。

03Cross-fitted nuisances

用完整 \(X\) 学习 propensity 与 outcome regressions。

04Orthogonal donor score

为每个 donor 得到 AIPW pseudo-outcome \(\widehat\Gamma_j\)。

05Belief neighborhood

由 belief similarity 构造 normalized \(\widehat\alpha_{ij}\)。

06Local effect

加权平均 donor scores,并报告 neighborhood diagnostics。

First estimator \[ \boxed{ \widehat\tau_{\mathrm{DiscoCATE},h}(x_i) = \sum_{j\in\mathcal I_i^{\mathrm{donor}}} \widehat\alpha_{ij}^{(h)}\widehat\Gamma_j } \]
Neighborhoodeffective donor sample size
Concentrationmaximum donor weight
Causal supporttreatment overlap
Geometrybelief-distance distribution

Identification ladder

它可识别吗?先问清楚“它”指什么

这篇论文最重要的理论贡献,是把 localized target、focal CATE 与 latent truth 分成三个问题。

  1. Level 1

    Ordinary CATE

    在 consistency、conditional exchangeability、positivity 与 integrability 下,\(\tau(x)\) 由 observed law 识别。Unit Abduction 不替代这些假设。

  2. Level 2

    Fixed-geometry localized effect

    对冻结且 pretreatment-measurable 的 \((\mathcal Q,d,K_h)\),首先识别的是:

    \[\theta_{h,\mathcal Q}(q)=\frac{\mathbb E[K_h\{d(q,Q_X)\}\{Y(1)-Y(0)\}]}{\mathbb E[K_h\{d(q,Q_X)\}]}.\]

    它是 soft-population effect;识别它不要求先识别真实 \(u_i\) 或真实 posterior。

  3. Level 3

    Shrinking-neighborhood limit

    在 localization regularity 下,\(\theta_{h,\mathcal Q}(q)\to\mathbb E[\tau(X)\mid Q_X=q]\)。这仍可能在 belief fiber 内平均不同的 \(X\)-values。

  4. Level 4

    When does it equal focal CATE?

    只有在 \(\tau(X)=g(Q_X)\),或满足 fiber equality / local injectivity 时,shrinking limit 才升级为 focal \(\tau(x_i)\)。

  5. Level 5

    Latent belief itself

    以上结果并不识别真实 \(u_i\)、唯一的 \(P(U\mid X=x)\)、latent coordinates、type-specific effect surface 或 cross-world individual-effect distribution。

最短结论:fixed-\(Q\) localized target 可以是 observed-law identifiable;它是否等于 focal CATE 是第二个、更强的问题;latent decomposition 是否真实则是第三个独立问题。

Claim boundary

这是一套明确、可证伪的 estimator interface,不是已经验证的 superiority claim

当前版本已经给出算法、fixed-map identification 和 falsification protocol;下一步决定价值的, 是在 matched conditions 下检验 complete belief 是否稳定改善 causal localization。

  • 不声称从单行 \((X,T,Y)\) 恢复真实 latent unit type。
  • 不声称 Unit Abduction 消除了未观测混杂。
  • 不声称 finite-bandwidth output 自动等于 focal CATE。
  • 不把 AIPW double robustness 延伸到错误或不稳定的 \(\widehat Q\)。
  • 不继承 Causal Forest 的 asymptotic normality 或 confidence intervals。
  • 不声称已优于 Causal Forest、R-/DR-learner 或 point-belief baselines。

Chinese question-led guide

建议先读中文问题导读,再按目标进入英文原稿

导读不是逐句翻译,而是用八个问题固定论文的直觉、公式、识别阶梯与证据边界。

推荐入口

《DiscoCATE》中文问题导读

适合第一次阅读、作者审阅与逐节反馈。与当前 manuscript v0 同步,不增加论文没有支持的 claim。

  1. 为什么 CATE estimation 首先是 neighborhood 问题?
  2. \(Q_i\) 是真实的 \(u_i\) 吗?
  3. 为什么不直接在 \(X\)-space 中找邻居?
  4. 两类 similarity 有什么区别?
  5. 为什么 similarity 和 AIPW score 必须分开?
  6. 论文真正识别了什么?
  7. “Doubly robust”是否意味着 abductor 错了也没关系?
  8. 什么实验结果才真正支持这条路线?
5 min

判断这篇论文在说什么

读“同一骨架、两个空间”→ 两条 similarity → identification ladder → claim boundary。

20 min

理解算法与理论主线

读 Abstract、Introduction、Setup、DiscoCATE-DR、Identification 4.1–4.3 与 Discussion。

Proof

审查理论

检查 fixed-\(\mathcal Q\) theorem、conditional double robustness、belief-fiber limit 与尚未完成的 learned-\(Q\) stability。

Falsify

判断路线是否值得继续

直接读 Evaluation Design:oracle-\(Q\)、multimodal weak evidence、matched baselines 与 leakage falsification。

Paper package

继续阅读、审查与反馈