OTA: 目标检测的最优传输分配
本文提出了一种新方法,使用不平衡最优输运(Unbalanced Optimal Transport)对目标检测模型进行训练,能够在 AP 和 AR 方面达到业内最好水平,并提供更快的初始收敛速度,适合于大规模模型的 GPU 实现。
Jul, 2023
以单一框架统一 Optimal Transport(OT)为基础的对抗方法,通过对统一框架的全面分析来阐明每个组成部分在训练动力学中的作用。我们提出了一个简单而新颖的方法,逐步改进生成分布,并逐渐与数据分布对齐。该方法在 CIFAR-10 上实现了 2.51 的 FID 得分,胜过了统一的基于 OT 的对抗方法。
Oct, 2023
本文介绍了一种新的方法 OTMatch,通过使用最优输运损失函数,利用类别之间的语义关系,提高半监督学习的性能。与现有最先进的方法 FreeMatch 相比,在 CIFAR-10、STL-10 和 ImageNet 数据集上分别实现了 3.18%、3.46% 和 1.28% 的错误率降低,证明了我们方法在半监督学习中利用语义关系提升学习性能的有效性和优越性。
Oct, 2023
我们提出了一种新的优化输运 (OT) 方法,称为课程和结构感知的优化输运 (CSOT),它同时考虑样本的全局和局部分布结构,构建了一个稳健的去噪和重新标记分配器。通过训练过程,分配器逐步为具有最高置信度的样本分配可靠的标签,这些标签具有全局可辨识性和局部一致性。我们的方法在学习噪声标签方面表现优于当前技术水平。
Dec, 2023
提出了一种新的利用最优传输理论来估计模型在无标签 out-of-distribution 数据上表现的方法,该方法称为置信度最优传输(COT),并且通过引入阈值处理的方法 Confidence Optimal Transport with Thresholding(COTT),可进一步提高精度。在各种标准基准测试中,实验表明 COT 和 COTT 优于现有的最先进方法,预测误差降低了高达 3 倍。
May, 2023
本文主要介绍了最优传输(OT)的基本概念及其在机器学习中的应用,提出了双重边界最优传输(DB-OT)方法,用于解决实际情况下涉及未确定目标的问题,并应用于聚类和长尾分类任务中,实验证明该方法在测试阶段具有良好的性能。
Jan, 2024
本文研究了使用少量副信息来学习代价函数的方法,该信息能够捕获到数据集中的子集对应关系,并提出了一种基于 Sinkhorn 算法的端对端优化器,实现了代价函数的自适应学习,结果表明该方法在图片、婚姻匹配和单细胞 RNA 测序等数据集上取得了明显的性能优势。
Sep, 2019