arXiv reaDer
グラフ ニューラル ネットワークを使用したオープンワールド キャリブレーションの学習
Learning for Open-World Calibration with Graph Neural Networks
私たちは、クラスタリングに表現のコンパクトさの尺度を組み込むことによって、オープンワールド認識のための閾値調整の問題に取り組みます。未知のものの発見と拒否に焦点を当てたオープンセット認識とは異なり、オープンワールド認識は、テスト時に共通化されていない未知のクラスに一般化可能な堅牢な表現を学習します。私たちが提案する方法は、次の 2 つの重要な観察に基づいています。(i) 高次元の視覚的埋め込み空間における隣接する画像間の表現構造は、オープンワールドへの移植性を促進するために利用できる強い自己相似性を持っています。(ii) クラス内の埋め込み構造は、真陽性率との相関はデータセット不変です。これらを動機として、私たちは、擬似ラベルと表現のコンパクトさを示す vMF 濃度を共同で予測するために、グラフ ニューラル ネットワーク (GNN) を中心とした統一フレームワークを設計します。これらの予測は、認識精度の統計的推定に変換できるため、距離のしきい値をより堅牢に校正して、オープンワールド クラスの目標有用性を達成できます。さまざまな視覚認識ベンチマークの結果は、オープンワールド、特に分布シフト下での従来の事後キャリブレーション方法よりも本発明の方法の優位性を実証しています。
We tackle the problem of threshold calibration for open-world recognition by incorporating representation compactness measures into clustering. Unlike the open-set recognition which focuses on discovering and rejecting the unknown, open-world recognition learns robust representations that are generalizable to disjoint unknown classes at test time. Our proposed method is based on two key observations: (i) representation structures among neighbouring images in high dimensional visual embedding spaces have strong self-similarity which can be leveraged to encourage transferability to the open world, (ii) intra-class embedding structures can be modeled with the marginalized von Mises-Fisher (vMF) probability, whose correlation with the true positive rate is dataset-invariant. Motivated by these, we design a unified framework centered around a graph neural network (GNN) to jointly predict the pseudo-labels and the vMF concentrations which indicate the representation compactness. These predictions can be converted into statistical estimations for recognition accuracy, allowing more robust calibration of the distance threshold to achieve target utility for the open-world classes. Results on a variety of visual recognition benchmarks demonstrate the superiority of our method over traditional posthoc calibration methods for the open world, especially under distribution shift.
updated: Fri May 19 2023 23:52:48 GMT+0000 (UTC)
published: Fri May 19 2023 23:52:48 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アソシエイト