Abbahaddou, Y., Ennadir, S., Lutzeyer, J. F., Malliaros, F. D., & Vazirgiannis, M. (2024). Post-Hoc Robustness Enhancement in Graph Neural Networks with Conditional Random Fields. arXiv preprint arXiv:2411.05399.
本研究は、学習済みグラフニューラルネットワーク(GNN)の、特にノード分類タスクにおける、敵対的な攻撃に対するロバスト性を、推論段階において向上させることを目的とする。
本研究では、GNNの入力データ間の類似性に基づいて条件付き確率場(CRF)を構築する。このCRFを用いて、GNNの元の予測を、ロバスト性の制約を満たすように修正する。具体的には、入力データとその近傍データに対する予測の類似性を維持するように、CRFを用いて予測を平滑化する。さらに、CRFの計算量を削減するために、グラフ構造とノード特徴の両方の空間においてサンプリング戦略を採用している。
RobustCRFは、GNNのロバスト性を向上させるための効果的な事後的な手法である。この手法は、モデルに依存せず、追加の学習を必要としないため、幅広いGNNモデルに適用できる。
本研究は、GNNのロバスト性を向上させるための新しい方向性を示唆している。RobustCRFは、実世界でGNNを安全に利用するための重要なステップとなる可能性がある。
他の言語に翻訳
原文コンテンツから
arxiv.org
深掘り質問