arXiv reaDer
覆面ジグソー パズル: ビジョン トランスフォーマーのための多彩な位置埋め込み
Masked Jigsaw Puzzle: A Versatile Position Embedding for Vision Transformers
ビジョン トランスフォーマー (ViT) に不可欠なコンポーネントである位置埋め込み (PE) は、多くのビジョン タスクで ViT のパフォーマンスを向上させることが示されています。ただし、PE は、入力パッチの空間情報が公開されているため、プライバシー漏洩のリスクが高い可能性があります。この警告は、精度、プライバシー、予測の一貫性などに対する PE の影響に関する一連の興味深い疑問を自然に提起します。これらの問題に取り組むために、Masked Jigsaw Puzzle (MJP) 位置埋め込み方法を提案します。特に、MJP は最初に、ブロックごとのランダム ジグソー パズル シャッフル アルゴリズムを介して選択されたパッチをシャッフルし、対応する PE をオクルードします。一方、オクルードされていないパッチの場合、PE は元のパッチのままですが、それらの空間的関係は、密な絶対ローカリゼーション リグレッサーによって強化されます。実験結果は、1) PE が 2D 空間関係を明示的にエンコードし、勾配反転攻撃の下で重大なプライバシー漏洩の問題を引き起こすことを明らかにしています。 2) 単純にシャッフルされたパッチを使用して ViT をトレーニングすると、問題を軽減できますが、精度が損なわれます。 3) 特定のシャッフル比の下で、提案された MJP は、大規模なデータセット (つまり、ImageNet-1K および ImageNet-C、-A/O) のパフォーマンスと堅牢性を向上させるだけでなく、典型的な勾配攻撃の下でのプライバシー保護能力も向上させます。大差で。ソース コードとトレーニング済みモデルは、~https://github.com/yhlleo/MJP で入手できます。
Position Embeddings (PEs), an arguably indispensable component in Vision Transformers (ViTs), have been shown to improve the performance of ViTs on many vision tasks. However, PEs have a potentially high risk of privacy leakage since the spatial information of the input patches is exposed. This caveat naturally raises a series of interesting questions about the impact of PEs on the accuracy, privacy, prediction consistency, etc. To tackle these issues, we propose a Masked Jigsaw Puzzle (MJP) position embedding method. In particular, MJP first shuffles the selected patches via our block-wise random jigsaw puzzle shuffle algorithm, and their corresponding PEs are occluded. Meanwhile, for the non-occluded patches, the PEs remain the original ones but their spatial relation is strengthened via our dense absolute localization regressor. The experimental results reveal that 1) PEs explicitly encode the 2D spatial relationship and lead to severe privacy leakage problems under gradient inversion attack; 2) Training ViTs with the naively shuffled patches can alleviate the problem, but it harms the accuracy; 3) Under a certain shuffle ratio, the proposed MJP not only boosts the performance and robustness on large-scale datasets (i.e., ImageNet-1K and ImageNet-C, -A/O) but also improves the privacy preservation ability under typical gradient attacks by a large margin. The source code and trained models are available at~https://github.com/yhlleo/MJP.
updated: Sun Mar 05 2023 11:04:12 GMT+0000 (UTC)
published: Wed May 25 2022 07:56:18 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アソシエイト