arXiv reaDer
制約付きディープネットワーク:ログバリア拡張によるラグランジュ最適化
Constrained Deep Networks: Lagrangian Optimization via Log-Barrier Extensions
この研究では、トレーニング中に畳み込みニューラルネットワーク(CNN)の出力に厳しい不等式制約を課すことを調査します。最近のいくつかの研究では、数百万のパラメーターを含む最新のCNNを処理する場合、単純なペナルティに対するラグランジュ最適化の理論的および実用的な利点は実際には実現されないことが示されています。したがって、制約付きCNNは通常、ペナルティ付きで処理されます。一連の制約のない損失を伴う制約付きCNN問題のラグランジュ最適化を近似する*ログバリア拡張*を提案します。標準的な内点法やログバリア法とは異なり、私たちの定式化は最初の実行可能な解決策を必要としません。提案された拡張により、双対性のギャップに上限が生じ、標準のログバリアの結果が一般化され、実行可能なソリューションの次善の証明書が得られます。非凸問題の準最適性は保証されませんが、この結果は、ログバリア拡張が暗黙の双対変数を介して制約付きCNNのラグランジュ最適化を近似する原理的な方法であることを示しています。さまざまな制約がある、弱く監視された画像セグメンテーション実験を報告します。これは、多数の制約を処理する場合に、精度、制約の満足度、トレーニングの安定性の点で、既存の制約付きCNNメソッドよりも大幅に優れていることを示しています。
This study investigates imposing hard inequality constraints on the outputs of convolutional neural networks (CNN) during training. Several recent works showed that the theoretical and practical advantages of Lagrangian optimization over simple penalties do not materialize in practice when dealing with modern CNNs involving millions of parameters. Therefore, constrained CNNs are typically handled with penalties. We propose *log-barrier extensions*, which approximate Lagrangian optimization of constrained-CNN problems with a sequence of unconstrained losses. Unlike standard interior-point and log-barrier methods, our formulation does not need an initial feasible solution. The proposed extension yields an upper bound on the duality gap -- generalizing the result of standard log-barriers -- and yielding sub-optimality certificates for feasible solutions. While sub-optimality is not guaranteed for non-convex problems, this result shows that log-barrier extensions are a principled way to approximate Lagrangian optimization for constrained CNNs via implicit dual variables. We report weakly supervised image segmentation experiments, with various constraints, showing that our formulation outperforms substantially the existing constrained-CNN methods, in terms of accuracy, constraint satisfaction and training stability, more so when dealing with a large number of constraints.
updated: Tue Mar 08 2022 19:53:58 GMT+0000 (UTC)
published: Mon Apr 08 2019 17:25:46 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト