跳到主内容

基础模型训练工作台

fm-adapt 双骨干 × 适配策略矩阵(PTB-XL)结果浏览 + 可复现训练命令生成。M1 只做配置与结果,不触发远程执行。

12
矩阵格数
0.9962
AF 最佳 AUROC
ECG-FM 预训练 · 全参微调
0.9474
MI 最佳 AUROC
ECG-FM 预训练 · Adapter
0.97%
LoRA 可训练参数占比
ECG-FM·AF,追平全参
范围(M1):数据源为已提交的 ml-training/benchmarks/fm-adapt-matrix-v1.json(从 gitignore 的 analysis/fm-adapt-matrix/runs/*/results.json 蒸馏)。 本页浏览 12 格矩阵、一键拼出与训练端同口径的 train_fm_adapt.py 命令(含 --seed 可复现)。 SSH 到 cqa1 远程拉起训练排到 M2——GPU 是临时机、今天不可测,且 deploy/research-cases/ 训练引擎属另一侧分区,不在 M1 触碰。
骨干策略AUROCF1@最优阈阈值最优轮可训练参数占比lr
从零训练 scratch
全参微调
0.89140.45370.3001,632,321100.00%0.001
从零训练 scratch
LoRA
0.84300.36530.151682,4335.00%0.001
从零训练 scratch
Adapter本格最优
0.94080.60920.4019132,7377.81%0.001
ECG-FM 预训练
全参微调本格最优
0.99620.92650.40391,474,433100.00%0.00005
ECG-FM 预训练
LoRA
0.99560.94230.403886,2730.97%0.0005
ECG-FM 预训练
Adapter
0.99300.93590.4063,102,8493.30%0.0005
训练命令生成器
python train_fm_adapt.py \
  --data-dir data/ptbxl-28/ \
  --output-dir runs/ecg-fm-lora-af \
  --backbone ecg-fm \
  --strategy lora \
  --task af \
  --epochs 20 \
  --batch-size 32 \
  --lr 0.0005 \
  --device cuda \
  --seed 42 \
  --save-model