PacketPatch/
├── README.md                          # プロジェクト概要（本ファイル）
├── LICENSE                            # MITライセンス
├── requirements.txt                   # Python依存関係
├── bert_base_config.json              # BERTモデル設定（12層、768次元、12ヘッド）
│
├── packet_adv/                        # ★ コア実験コード
│   ├── README.md                      # コアモジュールの使い方ガイド
│   ├── adv_fine_tuning_cls.py         # ステップ1: PatchGenerator（SpanBERT）の学習
│   ├── generate.py                    # ステップ2: 敵対的パケットの生成
│   └── gen_onnx.py                    # ステップ3（任意）: ONNX/TensorRT高速化
│
├── uer/                               # UER（Universal Encoder Representations）フレームワーク
│   ├── README.md                      # フレームワークドキュメント
│   ├── encoders/                      # エンコーダ実装（Transformer, RNN, CNN）
│   ├── layers/                        # ネットワーク層（Embedding, Attention, FFN）
│   ├── targets/                       # 学習ターゲット（MLM, NSP, Span-MBM など）
│   ├── models/                        # モデル定義
│   ├── utils/                         # ユーティリティ（トークナイザ、最適化、データロード）
│   ├── model_builder.py               # モデル構築
│   ├── model_loader.py                # 事前学習済み重みの読み込み
│   ├── model_saver.py                 # モデルチェックポイントの保存
│   ├── trainer.py                     # 学習ループ実装
│   └── opts.py                        # CLI引数定義
│
├── dataset/                           # データセットドキュメント
│   └── README.md                      # データセットの説明、ダウンロードリンク、前処理
│
├── image/                             # アーキテクチャと結果図
│   └── README.md                      # 図の説明
│
└── tools/                             # ユーティリティツール
    └── README.md                      # ツールの説明と使い方

# このリポジトリをクローン
git clone https://github.com/your-org/PacketPatch.git
cd PacketPatch

# 依存関係をインストール
pip install -r requirements.txt

cd packet_adv

# 最初に adv_fine_tuning_cls.py を編集:
#   - データセットパスを更新（47-50行目）
#   - 事前学習モデルパスを更新（73-75行目）
#   - 正しい num_classes を設定（56行目）
#   - 正しい学習セットサイズを設定（878行目）

python adv_fine_tuning_cls.py

cd packet_adv

# 最初に generate.py を編集:
#   - テストデータパスを更新（42-44行目）
#   - --load_model_path を学習済みチェックポイントに更新（68-69行目）
#   - BWO（帯域オーバーヘッド比率, 60行目, デフォルト0.1）を設定
#   - 'fun'パラメータで生成戦略を選択（50行目）

python generate.py

cd packet_adv

# 最初に gen_onnx.py を編集:
#   - モデルチェックポイントパスを更新（132行目）
#   - 出力パスを更新（145, 207行目）
#   - 精度モードを選択: 'fp16' または 'int8'（21行目）

python gen_onnx.py

@article{xu2025packetpatch,
    title = {PacketPatch: Practical Generation and Deployment of Adversarial Packets for Byte-Feature-Based Encrypted Traffic Classification},
    author = {Yuwei Xu and Yuanyuan Xu and Yunpeng Bai and Jiahui Chen and Kehui Song and Jie Cao and Qiao Xiang and Guang Cheng},
    journal = {Elsevier},
    year = {2025},
}