当前位置：首页 > news >正文

CDT-II：AI显微镜解码基因调控黑箱

news 2026/6/12 10:13:24

1. CDT-II：当AI显微镜遇见中心法则

在单细胞测序技术突飞猛进的今天，生物学家们面临着一个新的困境：海量的基因表达数据背后，隐藏着怎样的调控逻辑？传统AI模型虽然能做出准确预测，却像黑箱一样无法解释其内部机制。这就像拥有了一台能拍摄细胞照片的超高分辨率相机，却缺少解析这些图像含义的显微镜——直到Central Dogma Transformer II（CDT-II）的出现。

CDT-II的创新之处在于，它将分子生物学的中心法则直接映射到了神经网络架构中。想象一下，如果让Transformer模型像细胞处理遗传信息一样工作：DNA自注意力层模拟基因组序列的自我交互，RNA自注意力层捕捉基因间的共调控关系，而DNA-RNA交叉注意力层则重现转录调控过程。这种架构设计使得模型的"思考过程"变得透明可解释，每个注意力头都对应着特定的生物学功能。

2. 架构设计：从数据到洞察

2.1 模型的核心组件

CDT-II的架构包含三个关键模块，每个都对应中心法则的一个环节：

DNA编码器：采用Enformer预训练模型生成的基因组嵌入（896×3072矩阵），经过投影层降维至512维，再通过两层DNA自注意力捕捉115kb窗口内的基因组关系。这种设计使模型能够理解调控元件间的长程相互作用，比如增强子-启动子环。
RNA编码器：处理2,361个基因的单细胞表达数据。RawExpressionEncoder将log1p(CPM)标准化后的表达量与学习到的基因身份嵌入相结合，生成基因级表示。一层RNA自注意力识别共表达模块，例如参与相同通路的基因簇。
跨模态交互：DNA-RNA交叉注意力层（RNA作为query，DNA作为key/value）模拟转录调控过程。这个机制使模型能够回答诸如"当CTCF结合位点突变时，哪些基因的表达会发生变化"这类问题。

技术细节：所有注意力层使用8头缩放点积注意力，隐藏层维度2048，dropout率0.3。模型共约2100万参数，在单个NVIDIA A100 GPU上训练约2天。

2.2 数据处理的精妙之处

模型的输入输出设计体现了对生物学现实的深刻理解：

输入：仅需要原始单细胞表达数据（无需预先计算差异表达）和基因组序列嵌入。这种"raw data in, predictions out"的方式迫使模型自己学习什么是"表达变化"。
输出：预测log2倍数变化，直接对应实验生物学家的分析习惯。模型需要推断出例如"扰动位点A如何影响基因B"这样的调控关系。
训练技巧：采用ReduceLROnPlateau调度器（因子0.5，耐心10轮）和梯度裁剪（最大范数1.0），在噪声较大的单细胞数据上实现稳定训练。

3. 解码细胞的调控语言

3.1 注意力图谱的生物学解读

CDT-II最强大的功能在于其产生的注意力图谱可以直接对应已知的生物学现象：

GFI1B调控网络重建：RNA自注意力矩阵中，GFI1B（一种造血转录因子）的关注点集中在细胞周期调控基因上（如CDCA8、CDC20），与已知功能一致。Top100关注基因与实验确定的靶基因重叠达28个（6.6倍富集，P=3.5×10⁻¹⁷）。
RNA加工模块的发现：通过Louvain社区检测，RNA自注意力层和交叉注意力层独立识别出一个RNA剪接相关基因簇（80%基因重叠，P=9.3×10⁻⁴⁶）。这种"双重验证"增强了结果的可信度。
CTCF位点的富集：在28个测试基因中，交叉注意力显著富集于CTCF结合位点（平均7.67倍，P<0.001）。值得注意的是，模型仅接收一维序列信息就自动识别了这些三维基因组结构的关键锚点。

3.2 梯度分析：从表示到预测

虽然注意力机制揭示了模型的"思考过程"，但要量化调控强度还需要更精细的工具。CDT-II引入了梯度重要性评分：

计算雅可比矩阵Jji = ∂(输出基因j)/∂(输入基因i)，反映每个输入基因对预测结果的综合影响
对每个输入基因，计算其针对top100受影响基因的平均|Jji|作为重要性评分
与实验测得的log2FC比较，平均相关性达0.82

这种方法与传统"虚拟敲除"（直接设基因表达为0）形成鲜明对比，后者相关性仅约0.07。梯度分析的优势在于它探测的是模型在正常操作范围内的行为，更接近真实实验条件。

4. 从实验室到临床：TFRC案例研究

4.1 抗TfR1抗体的作用机制解析

TFRC（转铁蛋白受体1）是抗体药物PPMX-T003的靶点。CDT-II的梯度分析生成了一张全基因组调控图谱：

红细胞相关基因：EPB41和ACTR2的显著关联解释了临床试验中观察到的贫血副作用（血红蛋白持续下降）
铁依赖DNA合成：RRM2、RPA2等基因的扰动与网织红细胞减少相关
氧化应激模块：GCLM、MGST3等基因指向铁死亡机制（临床前研究已证实）
未报道的ER应激信号：PDIA6、SSR2等5个基因预测了蛋白质稳态紊乱，这尚未在临床报告中提及

4.2 方法学启示

这个案例展示了CDT-II作为"假设生成器"的价值：

已知效应验证模型准确性（5/10功能类别与临床一致）
新预测指导后续研究（如ER应激可作为安全性评估的新指标）
可扩展至其他药物靶点（CD52、CD44等）

5. 实操指南与经验分享

5.1 数据准备的关键

基因集筛选：初始使用9,335个基因时模型性能较差（r=0.37），而经过两个独立CRISPRi数据集验证的2,361个基因使r提升至0.64。这提示数据质量比模型容量更重要。
单细胞质量控制：仅保留明确分配的单扰动细胞（UMI≥50且无竞争信号），从60,505个细胞中筛选出15,657个高质量样本。

5.2 模型训练的技巧

学习率调度：采用ReduceLROnPlateau（因子0.5，耐心10轮）配合AdamW优化器（初始lr=1e-4），在单细胞数据的噪声中保持稳定训练。
正则化策略：权重衰减（1e-5）和dropout（p=0.3）防止过拟合，验证集表现与训练集几乎相同（r=0.65 vs 0.64）。