跳到正文
原文
arXiv cs.LG· Jiankai Huang, Hanwen Shen·· 19 天前AI 评分34

SE(3) 等变结构预测中动态 k-NN 图的精确拓扑通道调度:带阻尼置信域补偿

Refined Topology-Pass Scheduling: Damped Trust-Region Compensation for Dynamic k-NN Graphs in SE(3)-Equivariant Structure Prediction

AI 导读

研究者针对 SE(3) 等变结构预测提出了双通道拓扑调度方案,通过带阻尼置信域补偿动态 k-NN 图构建,在保证精度的同时降低逐块重建开销。该方案实现三种补偿级别,其中 L2 级别可完全复现逐块重建的边集,且整个过程取得 1.5x 的 Pass-1 与 1.76x 的 epoch 加速且无明显损失差距。相关运行时与实验代码已全部开源。

来源:arXiv cs.LG · arxiv.org