arXiv 机器学习· Jiankai Huang, Hanwen Shen·· 8 小时前AI 评分35
SE(3) 等变结构预测中动态 k-NN 图的精确拓扑 Pass 调度研究
Refined Topology-Pass Scheduling: Damped Trust-Region Compensation for Dynamic k-NN Graphs in SE(3)-Equivariant Structure Prediction
AI 导读
研究针对 SE(3) 等变结构预测提出了双 pass 拓扑调度与阻尼信赖域补偿方法，用于替代耗时的高频动态 kNN 图重构。该方案用 C++ 实现了 L1 至 L3 三种补偿级别，其中 L2 级别能完全精确复现每个边界的图边集。实测显示其 Pass-1 速度提升 1.5x、单 epoch 速度提升 1.76x 且无损失差距，相关运行时与实验代码已开源。
来源：arXiv 机器学习 · arxiv.org