arXiv reaDer
DatasetEquity: すべてのサンプルは同じように作成されていますか?データセット内の公平性の追求
DatasetEquity: Are All Samples Created Equal? In The Quest For Equity Within Datasets
データの不均衡は機械学習の分野ではよく知られた問題であり、データ収集のコスト、ラベル付けの難しさ、データの地理的分布が原因です。コンピュータ ビジョンでは、画像の外観によって引き起こされるデータ分布の偏りはまだほとんど解明されていません。クラス ラベルを使用したカテゴリ分布と比較すると、画像の外観により、クラス ラベルが提供するものを超えたオブジェクト間の複雑な関係が明らかになります。生のピクセルから抽出された深い知覚特徴をクラスタリングすると、データがより豊かに表現されます。この論文では、機械学習におけるデータの不均衡に対処するための新しい方法を紹介します。この方法では、深い知覚埋め込みとクラスタリングを使用して、画像の外観に基づいてサンプルの尤度を計算します。次に、提案された一般化焦点損失関数を使用したトレーニング中に、これらの可能性を使用してサンプルに異なる重み付けを行います。この損失は、深層学習アルゴリズムと簡単に統合できます。実験では、KITTI や nuScenes を含む自動運転ビジョン データセット全体でこの手法の有効性を検証します。損失関数は最先端の 3D オブジェクト検出方法を改善し、KITTI データセット内の過小評価されているクラス (Cyclist) で 200% を超える AP ゲインを達成します。結果は、この方法が一般化可能であり、既存の手法を補完し、小規模なデータセットやまれなクラスに対して特に有益であることを示しています。コードはhttps://github.com/towardsautonomy/DatasetEquityから入手できます。
Data imbalance is a well-known issue in the field of machine learning, attributable to the cost of data collection, the difficulty of labeling, and the geographical distribution of the data. In computer vision, bias in data distribution caused by image appearance remains highly unexplored. Compared to categorical distributions using class labels, image appearance reveals complex relationships between objects beyond what class labels provide. Clustering deep perceptual features extracted from raw pixels gives a richer representation of the data. This paper presents a novel method for addressing data imbalance in machine learning. The method computes sample likelihoods based on image appearance using deep perceptual embeddings and clustering. It then uses these likelihoods to weigh samples differently during training with a proposed Generalized Focal Loss function. This loss can be easily integrated with deep learning algorithms. Experiments validate the method's effectiveness across autonomous driving vision datasets including KITTI and nuScenes. The loss function improves state-of-the-art 3D object detection methods, achieving over 200% AP gains on under-represented classes (Cyclist) in the KITTI dataset. The results demonstrate the method is generalizable, complements existing techniques, and is particularly beneficial for smaller datasets and rare classes. Code is available at: https://github.com/towardsautonomy/DatasetEquity
updated: Sat Aug 19 2023 02:11:49 GMT+0000 (UTC)
published: Sat Aug 19 2023 02:11:49 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アソシエイト