MMApr, 2024

子模型多任务子集选择中的本地分布鲁棒性

TL;DR在本研究中,我们通过引入基于本地分布鲁棒性的参考分布邻域,提出了一种多任务子模块优化问题的方法。我们首先建议引入一个正则化项,利用相对熵与标准多任务目标之间的关系。然后通过对偶性证明这个新的公式本身等同于最大化一个子模块函数,这可以通过标准的贪婪选择方法高效实现。这种方法填补了多任务子集选择中性能 - 鲁棒性权衡优化的现有差距。为了数值验证我们的理论结果,我们在两种不同的情况下进行了测试,一种涉及对地球低轨道星座中的卫星进行选择以解决传感器选择问题,另一种涉及使用神经网络进行图像摘要任务。我们将我们的方法与两种其他算法进行了比较,这两种算法侧重于最坏情况下任务的性能优化和直接优化参考分布本身的性能。我们得出结论,我们的新公式提供了一种局部分布鲁棒且计算费用低廉的解决方案。