arXiv reaDer
RL-CoSeg:深層強化学習を備えた新しい画像コセグメンテーションアルゴリズム
RL-CoSeg : A Novel Image Co-Segmentation Algorithm with Deep Reinforcement Learning
本論文は深層強化学習(RL)に基づく自動画像コセグメンテーションアルゴリズムを提案した。既存のコセグメンテーションタスクは主に深層学習法に依存しており、取得された前景エッジは粗いことがよくあります。より正確な前景エッジを取得するために、ディープRLを使用してこの問題を解決し、より細かいセグメンテーションを実現します。私たちの知る限り、これはRLメソッドをコセグメンテーションに適用する最初の作業です。問題をマルコフ決定過程(MDP)として定義し、非同期アドバンテージアクタークリティカル(A3C)を使用してRLによって最適化します。 RL画像コセグメンテーションネットワークは、画像間の相関関係を使用して、関連する画像のセットから一般的なオブジェクトと顕著なオブジェクトをセグメント化します。自動セグメンテーションを実現するために、RL-CoSegメソッドはユーザーのヒントを排除します。画像のコセグメンテーション問題については、A3Cモデルに基づく協調RLアルゴリズムを提案します。コセグメンテーションのための画像のコアテンションを得るために、シャムRLコセグメンテーションネットワーク構造を提案します。自動RLアルゴリズムの自己注意を改善して、長距離依存性を取得し、受容野を拡大します。自己注意によって得られた画像の特徴情報は、削除されたユーザーのヒントを補足し、より正確なアクションを取得するのに役立ちます。実験結果は、私たちの方法が粗い初期セグメンテーションと細かい初期セグメンテーションの両方で効果的にパフォーマンスを改善できることを示しており、インターネットデータセット、iCosegデータセット、MLMR-COSデータセットで最先端のパフォーマンスを実現しています。
This paper proposes an automatic image co-segmentation algorithm based on deep reinforcement learning (RL). Existing co-segmentation tasks mainly rely on deep learning methods, and the obtained foreground edges are often rough. In order to obtain more precise foreground edges, we use deep RL to solve this problem and achieve the finer segmentation. To our best knowledge, this is the first work to apply RL methods to co-segmentation. We define the problem as a Markov Decision Process (MDP) and optimize it by RL with asynchronous advantage actor-critic (A3C). The RL image co-segmentation network uses the correlation between images to segment common and salient objects from a set of related images. In order to achieve automatic segmentation, our RL-CoSeg method eliminates user's hints. For the image co-segmentation problem, we propose a collaborative RL algorithm based on the A3C model. We propose a Siamese RL co-segmentation network structure to obtain the co-attention of images for co-segmentation. We improve the self-attention for automatic RL algorithm to obtain long-distance dependence and enlarge the receptive field. The image feature information obtained by self-attention can be used to supplement the deleted user's hints and help to obtain more accurate actions. Experimental results have shown that our method can improve the performance effectively on both coarse and fine initial segmentations, and it achieves the state-of-the-art performance on Internet dataset, iCoseg dataset and MLMR-COS dataset.
updated: Tue Apr 12 2022 17:06:15 GMT+0000 (UTC)
published: Tue Apr 12 2022 17:06:15 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アソシエイト