arXiv reaDer
フェデレーテッド インクリメンタル セマンティック セグメンテーション
Federated Incremental Semantic Segmentation
フェデレーテッド ラーニング ベースのセマンティック セグメンテーション (FSS) は、ローカル クライアントでの分散型トレーニングを通じて広く注目を集めています。ただし、ほとんどの FSS モデルは、カテゴリが事前に固定されていることを前提としているため、古いクラスにアクセスするためのメモリ ストレージがなくても、ローカル クライアントが新しいカテゴリを段階的に受け取る実際のアプリケーションでは、古いカテゴリが大幅に忘れられます。さらに、新しいクラスを収集する新しいクライアントは、FSS のグローバル トレーニングに参加する可能性があり、壊滅的な忘却をさらに悪化させます。上記の課題を克服するために、クライアント内とクライアント間の両方の側面から古いクラスの異種忘却に対処するための Forgetting-Balanced Learning (FBL) モデルを提案します。具体的には、適応クラスバランス疑似ラベル付けによって生成された疑似ラベルのガイダンスの下で、忘却バランスセマンティック補償損失と忘却バランス関係一貫性損失を開発して、バックグラウンドシフトを伴う古いカテゴリのクライアント内異種忘却を修正します。ローカル クライアント内でバランスの取れた勾配の伝播と関係の一貫性の蒸留を実行します。さらに、クライアント間の観点から異種忘却に取り組むために、タスク遷移モニターを提案します。プライバシー保護の下で新しいクラスを識別し、関係蒸留のために最新の古いグローバル モデルを保存できます。定性的実験により、比較方法に対するモデルの大幅な改善が明らかになりました。コードは https://github.com/JiahuaDong/FISS で入手できます。
Federated learning-based semantic segmentation (FSS) has drawn widespread attention via decentralized training on local clients. However, most FSS models assume categories are fixed in advance, thus heavily undergoing forgetting on old categories in practical applications where local clients receive new categories incrementally while have no memory storage to access old classes. Moreover, new clients collecting novel classes may join in the global training of FSS, which further exacerbates catastrophic forgetting. To surmount the above challenges, we propose a Forgetting-Balanced Learning (FBL) model to address heterogeneous forgetting on old classes from both intra-client and inter-client aspects. Specifically, under the guidance of pseudo labels generated via adaptive class-balanced pseudo labeling, we develop a forgetting-balanced semantic compensation loss and a forgetting-balanced relation consistency loss to rectify intra-client heterogeneous forgetting of old categories with background shift. It performs balanced gradient propagation and relation consistency distillation within local clients. Moreover, to tackle heterogeneous forgetting from inter-client aspect, we propose a task transition monitor. It can identify new classes under privacy protection and store the latest old global model for relation distillation. Qualitative experiments reveal large improvement of our model against comparison methods. The code is available at https://github.com/JiahuaDong/FISS.
updated: Mon Apr 10 2023 14:34:23 GMT+0000 (UTC)
published: Mon Apr 10 2023 14:34:23 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アソシエイト