适应度预测模型

在同一套 138 维特征(one-hot 80 + 理化描述符 28 + 位点对交互 30)上训练五个 baseline, 按验证集 Spearman 选优。核心挑战是外推:模型只见过 ≤2 点突变,却要给 3–4 点组合排序。

最终选用:xgboost 验证集 Spearman 0.902 测试集 Spearman 0.489 测试集 recall@100 0.19
测试集 Spearman ρ
0
对 147,193 个未知高阶组合排序
测试集 Pearson r
0
log 空间的线性相关性
测试集 MSE
0
log10(fitness + 0.01) 空间
验证集 Spearman ρ
0
低阶变体的留出验证表现

五个 baseline 的对比

验证集(内插)与测试集(外推)之间的落差,就是这个任务的真实难度

Spearman 相关系数

内插 vs 外推

Top-k 命中率(测试集)

预测 Top-k 中有多少确实属于真实 Top-k

完整指标表

按验证集 Spearman 降序排列,第一行为最终选用的模型
模型 验证 ρ验证 MSE 测试 ρ测试 r 测试 MSE测试 MAE测试 R² recall@10recall@100NDCG@100
XGBoost 梯度提升树 选用 0.9025 0.0470 0.4893 0.7047 0.1552 0.2871 0.2889 0.000 0.190 0.927
极端随机树 0.8971 0.0523 0.4625 0.7220 0.1257 0.2528 0.4244 0.000 0.150 0.906
随机森林 0.8911 0.0653 0.4562 0.6439 0.1633 0.2962 0.2522 0.000 0.030 0.889
多层感知机 MLP 0.8579 0.0940 0.4586 0.6292 0.2460 0.3937 -0.1266 0.000 0.110 0.898
Ridge 线性回归 0.8505 0.1491 0.4720 0.5247 0.6482 0.6276 -1.9690 0.000 0.020 0.869
📌
所有模型都呈现同一个规律:验证集 ρ ≈ 0.9,测试集 ρ ≈ 0.5。 训练集主要包含单点和双点突变,测试集包含三点和四点突变,存在分布差异和高阶相互作用。 训练集最大值并非所有回归模型的预测上限。高适应度区域的误差说明需要评估探索策略, 但仅凭预测误差不能证明 Agent 优于模型直接推荐。

可视化诊断

预测质量、能力边界与特征重要性

模型对比

模型对比

五个 baseline 的 Spearman / MSE / Top-k 命中率

预测质量

预测质量

内插与外推的散点密度图及 Top-k 命中率曲线

特征重要性

特征重要性

模型最依赖的特征与四个位点的贡献占比

批次预测误差

批次预测误差

所有送检候选的预测值与虚拟实验真实值对照