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/                        # 网络层（嵌入、注意力、前馈网络）
│   ├── targets/                       # 训练目标（MLM、NSP、Span-MBM 等）
│   ├── models/                        # 模型定义
│   ├── utils/                         # 实用工具（分词器、优化器、数据加载）
│   ├── model_builder.py               # 模型构建
│   ├── model_loader.py                # 预训练权重加载
│   ├── model_saver.py                 # 模型检查点保存
│   ├── trainer.py                     # 训练循环实现
│   └── opts.py                        # 命令行参数定义
│
├── 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},
}