Giới thiệu dự án

Trong kỷ nguyên chuyển đổi số và bùng nổ hạ tầng điện toán đám mây, các cuộc tấn công mạng ngày càng gia tăng về quy mô lẫn mức độ tinh vi. Theo thống kê an ninh mạng toàn cầu, các cuộc tấn công từ chối dịch vụ phân tán (DDoS), xâm nhập có chủ đích (APT) và phần mềm độc hại mới tăng hơn 38% mỗi năm. Để bảo vệ hạ tầng mạng, hệ thống phát hiện xâm nhập (Intrusion Detection System - IDS) đóng vai trò là tuyến phòng thủ then chốt giúp giám sát lưu lượng, nhận diện bất thường và đưa ra cảnh báo sớm cho trung tâm điều hành an ninh mạng (SOC).

+-----------------------------------------------------------------------------------+
|                        MÔ HÌNH TỔNG QUAN HỆ THỐNG PHÒNG THỦ IDS                   |
+-----------------------------------------------------------------------------------+
|                                                                                   |
|  [ Lưu lượng mạng / NetFlow ]                                                     |
|  [ Tiền xử lý & Xây dựng Đồ thị G(V, E) ]                                         |
|             [ Đánh giá Tính bền vững & Né tránh ]                                 |
|             [ Phòng thủ: Tái huấn luyện đối kháng ]                               |
|             [ Triển khai thực nghiệm Mạng SDN (Mininet + Ryu) ]                   |
|                                                                                   |
+-----------------------------------------------------------------------------------+

Vấn đề nghiên cứu và Thách thức cốt lõi

Mặc dù các mô hình Học sâu (Deep Learning - DL) như Mạng nơ-ron tích chập (Convolutional Neural Network - CNN) hay Bộ tự mã hóa biến phân (Variational Autoencoder - VAE) đã đạt độ chính xác cao trong phân loại luồng mạng tĩnh, chúng bỏ qua mối quan hệ tương tác liên kết giữa các thực thể IP/Port trong không gian phi Euclid (Non-Euclidean Space). Mạng nơ-ron đồ thị (Graph Neural Network - GNN) nổi lên như giải pháp đột phá giúp mô hình hóa topology mạng tổng thể.

Tuy nhiên, các mô hình Học máy và GNN đều đối mặt với lỗ hổng nghiêm trọng trước Tấn công đối kháng (Adversarial Attacks):

  • Tấn công né tránh (Evasion Attacks): Kẻ tấn công biến đổi thuộc tính gói tin hoặc can thiệp topology đồ thị (thêm/xóa liên kết luồng) trong pha kiểm thử (inference phase) để qua mặt bộ phát hiện mà không làm mất tính độc hại của cuộc tấn công.
  • Tính chuyển giao (Adversarial Transferability): Các mẫu đối kháng sinh ra từ một mô hình hộp trắng đơn giản (như CNN hoặc VAE) có thể đánh lừa thành công một mô hình hoàn toàn khác biệt về cấu trúc như GNN (Black-box Attack).

Mục tiêu nghiên cứu cụ thể

  1. Thiết kế và tối ưu mô hình E-GraphSAGE (Edge-featured GraphSAGE): Mở rộng thuật toán GraphSAGE truyền thống nhằm nhúng trực tiếp vector thuộc tính cạnh luồng mạng (NetFlow features) để phát hiện xâm nhập.
  2. Xây dựng kịch bản tấn công né tránh cấu trúc: Đánh giá định lượng mức độ suy giảm hiệu năng của GNN khi bị làm nhiễu cấu trúc liên kết thông qua thao tác thêm/xóa 5% và 10% tổng số cạnh đồ thị.
  3. Đánh giá tính chuyển giao đối kháng liên kiến trúc: Sử dụng công cụ Adversarial Robustness Toolbox (ART) để tạo mẫu tấn công trên CNN, VAE và chuyển giao sang mô hình E-GraphSAGE trên hai tập dữ liệu chuẩn NF-UNSW-NB15 và NF-CSE-CIC-IDS2018.
  4. Phát triển chiến lược phòng thủ và tái huấn luyện (Adversarial Retraining): Tăng cường tính bền vững (robustness) cho IDS và kiểm chứng cơ chế thực thi trên mạng khả lập trình (Software-Defined Networking - SDN).

Phạm vi và Giới hạn đề tài

  • Phạm vi dữ liệu: Thực nghiệm trên 2 bộ dữ liệu chuẩn hóa dạng NetFlow gồm NF-UNSW-NB15 và NF-CSE-CIC-IDS2018.
  • Phạm vi mô hình: So sánh đối chiếu 3 kiến trúc: CNN, VAE và E-GraphSAGE.
  • Môi trường mô phỏng: Triển khai kiến trúc mạng khả lập trình SDN dựa trên Mininet và bộ điều khiển OpenFlow/Ryu Controller.

Phân tích và thiết kế giải pháp

Phân tích hiện trạng

Tiêu chí phân tích IDS dựa trên Chữ ký (Snort/Suricata) IDS Học sâu truyền thống (CNN / MLP / VAE) IDS Đồ thị đề xuất (E-GraphSAGE IDS)
Khả năng bắt Zero-day Kém (chỉ nhận diện mẫu đã biết) Tốt (dựa trên phân tích bất thường) Xuất sắc (bắt bất thường hành vi & tương tác topo)
Khai thác quan hệ Topo Không hỗ trợ Không hỗ trợ (xem từng flow độc lập) Toàn diện (tổng hợp lân cận đa bước nhảy)
Kháng nhiễu đối kháng Trung bình (dễ bị obfuscation chuỗi) Rất kém (dễ vỡ trước FGSM, PGD, CW) Bền vững hơn sau khi tái huấn luyện đối kháng
Chi phí tính toán Thấp ($O(1)$ lookup) Trung bình ($O(N)$ inference) $O(V + E)$ theo kích thước đồ thị con

Yêu cầu hệ thống theo mô hình MoSCoW

  • Must Have: Mô hình E-GraphSAGE nhúng thuộc tính cạnh luồng mạng; thuật toán sinh mẫu đối kháng gradient; pipeline tái huấn luyện.
  • Should Have: Module chuyển giao mẫu đối kháng từ CNN/VAE sang GNN; cơ chế giảm chiều dữ liệu NetFlow tự động.
  • Could Have: Tích hợp module can thiệp luồng động trên SDN Controller (Ryu) khi phát hiện tấn công đối kháng.
  • Won't Have: Triển khai phần cứng ASIC chuyên dụng trong pha thử nghiệm hiện tại.

Thiết kế hệ thống và Kiến trúc kỹ thuật

Hệ thống được thiết kế theo kiến trúc module hóa phân tầng:

[ Thu thập Dữ liệu NetFlow (v9/IPFIX) ]
[ Mô hình Đối kháng (ART) ]  [ Pipeline E-GraphSAGE ]
- FGSM, PGD, CW              - Aggregate: AGG_k(e_uv)
- Trích xuất Perturbation    - Edge Embedding Update
[ Module Đánh giá Né tránh & Chuyển giao ]
[ SDN Controller & Phòng thủ Tái huấn luyện ]

Tech Stack và Phiên bản công nghệ

  • Ngôn ngữ cốt lõi: Python v3.9.16
  • Thư viện Học sâu & Đồ thị: PyTorch v2.0.1, PyTorch Geometric (PyG) v2.3.1
  • Bộ công cụ Sinh đối kháng: Adversarial Robustness Toolbox (IBM ART) v1.15.2
  • Thư viện Khoa học dữ liệu: Scikit-Learn v1.2.2, NetworkX v3.1, Pandas v2.0.3
  • Môi trường SDN: Mininet v2.3.0, OpenFlow Protocol v1.3, Ryu SDN Controller v4.34

Implementation và kết quả

Chi tiết thuật toán E-GraphSAGE và Sinh mẫu đối kháng

Thuật toán E-GraphSAGE cải tiến giải quyết bài toán biểu diễn luồng mạng bằng cách cập nhật trạng thái đỉnh dựa trên vector thuộc tính cạnh $e_{uv}^{(k-1)}$ của các kết nối lân cận $\mathcal{N}(v)$:

$$h_{\mathcal{N}(v)}^{(k)} = \text{AGG}k\left(\left{ e{uv}^{(k-1)}, \forall u \in \mathcal{N}(v), uv \in E \right}\right)$$

$$h_v^{(k)} = \sigma\left( W^{(k)} \cdot \text{CONCAT}\left( h_v^{(k-1)}, h_{\mathcal{N}(v)}^{(k)} \right) \right)$$

Sau $K$ bước nhảy (hops), biểu diễn nhúng của cạnh $uv$ được tổng hợp trực tiếp từ hai đỉnh đầu mút:

$$z_{uv} = \text{CONCAT}\left( z_u^{(K)}, z_v^{(K)} \right)$$

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch_geometric.nn import MessagePassing

class EGraphSAGEConv(MessagePassing):
    def __init__(self, in_channels, edge_channels, out_channels):
        super(EGraphSAGEConv, self).__init__(aggr='mean')
        self.lin_node = nn.Linear(in_channels, out_channels)
        self.lin_edge = nn.Linear(edge_channels, out_channels)
        self.lin_update = nn.Linear(out_channels * 2, out_channels)

    def forward(self, x, edge_index, edge_attr):
        # x: [num_nodes, in_channels], edge_attr: [num_edges, edge_channels]
        return self.propagate(edge_index, x=x, edge_attr=edge_attr)

    def message(self, x_j, edge_attr):
        # Kết hợp thông tin node lân cận và đặc trưng thuộc tính cạnh
        msg = F.relu(self.lin_edge(edge_attr))
        return msg

    def update(self, aggr_out, x):
        h_v = self.lin_node(x)
        concat_h = torch.cat([h_v, aggr_out], dim=-1)
        return F.relu(self.lin_update(concat_h))

Phương pháp sinh mẫu đối kháng dựa trên tối ưu hóa gradient (Fast Gradient Sign Method - FGSM):

$$x_{adv} = x + \epsilon \cdot \text{sign}\left( \nabla_x \mathcal{L}(\theta, x, y) \right)$$

Trong đó $\epsilon$ là hệ số nhiễu giới hạn ($L_\infty$-norm budget), $\mathcal{L}$ là hàm mất mát phân loại (Cross-Entropy Loss).

from art.attacks.evasion import FastGradientMethod, ProjectedGradientDescent
from art.estimators.classification import PyTorchClassifier

# Khởi tạo bộ ước lượng ART cho mô hình Baseline
classifier = PyTorchClassifier(
    model=cnn_model,
    loss=nn.CrossEntropyLoss(),
    optimizer=torch.optim.Adam(cnn_model.parameters(), lr=0.001),
    input_shape=(1, feature_dim),
    nb_classes=2,
    clip_values=(0.0, 1.0)
)

# Sinh mẫu tấn công đối kháng Evasion bằng PGD
pgd_attacker = ProjectedGradientDescent(
    estimator=classifier,
    eps=0.1,
    eps_step=0.01,
    max_iter=40,
    targeted=False
)
x_test_adv = pgd_attacker.generate(x=x_test_clean)

Testing và Validation

Thực nghiệm được thực hiện trên 2 bộ dữ liệu chuẩn NetFlow:

  1. NF-UNSW-NB15: Bao gồm 1,623,118 bản ghi luồng mạng với 43 thuộc tính số và nhãn tấn công đa dạng (DoS, Exploits, Generic, Reconnaissance).
  2. NF-CSE-CIC-IDS2018: Tập dữ liệu quy mô lớn gồm 18,893,708 bản ghi luồng mạng mô phỏng hạ tầng doanh nghiệp AWS.

1. Kết quả tấn công né tránh cấu trúc (Thao tác Thêm/Xóa cạnh)

Khi áp dụng phương pháp can thiệp topology đồ thị bằng cách thêm/xóa ngẫu nhiên hoặc có chủ đích 5% và 10% số cạnh kết nối trong đồ thị E-GraphSAGE:

Tập dữ liệu Kịch bản thực nghiệm Precision (%) Recall (%) F1-Score (%) Accuracy (%)
NF-UNSW-NB15 Gốc (Clean Graph) 96.82 95.40 96.10 96.25
Thêm/Xóa 5% cạnh 84.15 81.30 82.70 83.10
Thêm/Xóa 10% cạnh 72.40 68.90 70.61 71.35
NF-CSE-CIC-IDS2018 Gốc (Clean Graph) 98.45 97.80 98.12 98.20
Thêm/Xóa 5% cạnh 87.60 84.25 85.89 86.40
Thêm/Xóa 10% cạnh 75.10 71.60 73.31 74.50

Nhận xét: Việc thao tác 10% cạnh làm suy giảm nghiêm trọng chỉ số F1-Score của GNN (giảm tới ~25.49% trên NF-UNSW-NB15), chứng minh rằng các mô hình GNN dù mạnh mẽ vẫn cực kỳ nhạy cảm trước nhiễu loạn liên kết topo.

2. Kết quả Chuyển giao Đối kháng (Adversarial Transferability)

Mẫu đối kháng được sinh từ mô hình CNN và VAE thông qua kỹ thuật FGSM/PGD, sau đó nạp trực tiếp vào mô hình mục tiêu E-GraphSAGE (chưa từng thấy mẫu này trong pha train):

Mô hình Nguồn Mô hình Mục tiêu Tập dữ liệu Tỷ lệ né tránh thành công (ASR %) F1-Score còn lại của GNN (%)
CNN E-GraphSAGE NF-UNSW-NB15 64.30% 58.20%
VAE E-GraphSAGE NF-UNSW-NB15 59.80% 62.45%
CNN E-GraphSAGE NF-CSE-CIC-IDS2018 68.75% 54.10%
VAE E-GraphSAGE NF-CSE-CIC-IDS2018 62.10% 60.80%

3. Kết quả Tái huấn luyện Đối kháng (Adversarial Retraining)

Sau khi áp dụng phương pháp tái huấn luyện bằng cách đưa 20% mẫu nhiễu topo và mẫu đối kháng chuyển giao vào tập huấn luyện mới:

Kịch bản kiểm thử (NF-UNSW-NB15) F1-Score Trước Re-training (%) F1-Score Sau Re-training (%) Tỷ lệ phục hồi hiệu năng
Dữ liệu sạch (Clean Data) 96.10% 95.85% Giữ vững (-0.25%)
Nhiễu 5% cạnh liên kết 82.70% 93.40% +10.70%
Mẫu chuyển giao từ CNN 58.20% 89.65% +31.45%
Mẫu chuyển giao từ VAE 62.45% 91.20% +28.75%

Đổi mới và đóng góp

  1. Kiến trúc E-GraphSAGE tối ưu hóa cho NetFlow: Khác với GraphSAGE chuẩn chỉ xử lý thuộc tính nút, kiến trúc đề xuất cập nhật trạng thái đỉnh thông qua toán tử tổng hợp trực tiếp trên thuộc tính luồng mạng (cạnh), giúp giảm chi phí tính toán đồ thị trong mạng lớn.
  2. Khám phá định lượng về tính chuyển giao liên kiến trúc (Cross-Architecture Transferability): Chứng minh bằng thực nghiệm rằng vector gradient đối kháng sinh ra từ các mô hình học sâu truyền thống (CNN, VAE) hoàn toàn có khả năng vô hiệu hóa mô hình học sâu cấu trúc đồ thị (GNN) với tỷ lệ vượt qua lên tới 68.75%.
  3. Chiến lược Phòng thủ Tái huấn luyện Tích hợp (Adversarial Robust Retraining): Đề xuất pipeline tái cấu trúc ma trận kề và vector đặc trưng giúp phục hồi hơn 31% hiệu năng phân loại của hệ thống IDS trước các biến thể tấn công zero-day.
  4. Mô hình hóa kiểm chứng trên môi trường SDN thực tế: Kết nối luồng phân tích từ GNN trực tiếp với Ryu SDN Controller, cho phép tự động cô lập cổng và đẩy FlowMod rules chặn luồng tấn công đối kháng theo thời gian thực.

Ứng dụng thực tế và triển khai

Kịch bản Triển khai Trung tâm Điều hành An ninh Mạng (SOC Enterprise)

Mô hình E-GraphSAGE được đóng gói thành Microservice phân tích bất thường luồng mạng, tích hợp trong hệ sinh thái giám sát SDN của doanh nghiệp viễn thông hoặc ngân hàng:

[ Switches OpenFlow (ToR) ]
[ Logstash / Apache Kafka Message Bus ]
[ E-GraphSAGE Inference Engine (Triton / PyTorch C++) ]
[ Ryu Controller ] [ Forwarding ]
[ Chặn IP/Port tức thì ]

Yêu cầu Tài nguyên Hệ thống

  • Hạ tầng máy chủ suy luận: CPU Intel Xeon Gold 6248R (8 Cores), 32GB RAM, GPU NVIDIA RTX A4000 (16GB VRAM) hoặc Tesla T4.
  • Băng thông xử lý: Đạt thông lượng 45,000 flows/giây với độ trễ suy luận trung bình $\le 18.5\text{ms}$.
  • Ước tính Hiệu quả Đầu tư (ROI): Giảm thiểu 65% thời gian điều tra sự cố của kỹ sư SOC (MTTR) và ngăn chặn nguy cơ tổn thất trung bình 2.5 triệu USD do các cuộc tấn công né tránh gây tê liệt hệ thống.

Hạn chế và hướng phát triển

Hạn chế kỹ thuật hiện tại

  • Độ phức tạp tính toán đồ thị thời gian thực: Việc cập nhật và tái tạo cấu trúc đồ thị động liên tục trên mạng có hàng triệu kết nối đồng thời đòi hỏi dung lượng bộ nhớ GPU đáng kể.
  • Ranh giới gán nhãn luồng: Các luồng mạng có thời gian tồn tại cực ngắn (mice flows) có thể tạo ra các đồ thị con thưa thớt, làm giảm hiệu quả tổng hợp lân cận của phép toán AGG.

Hướng phát triển tương lai

  • Dynamic Temporal GNN: Ứng dụng Mạng nơ-ron đồ thị động theo chuỗi thời gian (T-GNN) để nắm bắt biến thiên topology theo từng mili-giây.
  • Phòng thủ Tự động bằng Học tăng cường (Reinforcement Learning): Tích hợp tác tử RL trên SDN Controller để tự động tối ưu hóa chính sách thay đổi địa chỉ IP ảo (Moving Target Defense - MTD) nhằm vô hiệu hóa nỗ lực thăm dò của kẻ tấn công đối kháng.

Đối tượng hưởng lợi

  • Sinh viên & Học viên cao học: Tiếp cận tài liệu tham khảo chuẩn mực về An toàn thông tin, toán học đối kháng trong AI, kỹ thuật triển khai PyTorch Geometric và mô phỏng SDN.
  • Kỹ sư An ninh mạng & AI/MLOps Engineers: Nắm vững phương pháp kiểm thử độ bền (stress-testing) của mô hình phân loại trước các cuộc tấn công Adversarial Evasion và phương pháp tái huấn luyện thực chiến.
  • Doanh nghiệp & Đơn vị Quản trị Mạng: Sở hữu thiết kế kiến trúc IDS thế hệ mới có khả năng tích hợp linh hoạt vào hạ tầng SDN hiện hữu, nâng cao năng lực bảo vệ dữ liệu trọng yếu.
  • Cộng đồng Nghiên cứu An ninh AI: Kế thừa kết quả thực nghiệm định lượng về tính chuyển giao đối kháng trên các bộ dữ liệu NetFlow tiêu chuẩn.

Câu hỏi thường gặp

1. Yêu cầu kỹ thuật tối thiểu để triển khai mô hình E-GraphSAGE IDS là gì?

Hệ thống yêu cầu máy chủ Linux (Ubuntu 20.04/22.04 LTS), cài đặt Python 3.9+, PyTorch 2.0+, CUDA 11.8+, tối thiểu 16GB RAM và 1 GPU có VRAM $\ge 8\text{GB}$ để đáp ứng tốc độ dựng đồ thị luồng và suy luận liên tục.

2. Giới hạn khả năng mở rộng (Scalability) của mô hình đồ thị và giải pháp khắc phục?

Khi số lượng host và luồng mạng vượt quá hàng triệu node/cạnh, đồ thị toàn cục sẽ gặp hiện tượng quá tải bộ nhớ. Giải pháp khắc phục là sử dụng cơ chế Mini-batch Subgraph Sampling của GraphSAGE kết hợp cụm phân tán DGL/PyG để chỉ lấy mẫu $k$-hàng xóm quan trọng nhất thay vì xử lý toàn bộ đồ thị.

3. Mô hình có thể tích hợp với các hệ thống SIEM/SOC hiện hữu như thế nào?

E-GraphSAGE xuất kết quả phân loại dưới định dạng chuẩn JSON qua Kafka Topic hoặc Syslog RFC 5424. Các nền tảng SIEM như Splunk, IBM QRadar hay Elastic Security có thể tiêu thụ log này trực tiếp để kích hoạt playbook xử lý sự cố tự động.

4. Chi phí và chu kỳ bảo trì hệ thống định kỳ như thế nào?

Mô hình cần được cập nhật trọng số định kỳ (thường là 1-2 tuần/lần) thông qua pipeline Tái huấn luyện đối kháng tự động với các mẫu tấn công mới thu thập từ Honeypot. Chi phí duy trì chủ yếu là điện năng và khấu hao phần cứng máy chủ GPU.

5. Thời gian hoàn vốn đầu tư (ROI) ước tính cho doanh nghiệp?

Với việc giảm thiểu 80% tỷ lệ cảnh báo sai (False Positives) và ngăn chặn thành công các biến thể tấn công né tránh, thời gian hoàn vốn đầu tư ước tính đạt từ 6 đến 9 tháng sau khi đưa vào vận hành chính thức tại Trung tâm SOC.


Kết luận

Đồ án khóa luận tốt nghiệp "A Study on the Evasiveness and Transferability of Adversarial Attacks against Intrusion Detection Systems" của nhóm tác giả Lê Minh Phúc và Mai Anh Khoa (Trường ĐH Công nghệ Thông tin - ĐHQG TP.HCM) dưới sự hướng dẫn của ThS. Phan Thế Duy đã giải quyết thành công bài toán đánh giá toàn diện tính bền vững của các hệ thống IDS thế hệ mới. Đề tài không chỉ chứng minh các lỗ hổng tiềm tàng của mô hình GNN trước các kỹ thuật can thiệp topo và mẫu đối kháng chuyển giao từ CNN/VAE, mà còn cung cấp giải pháp phòng thủ tái huấn luyện vững chắc và mô hình triển khai thực tế trên mạng SDN. Đây là đóng góp khoa học và thực tiễn giá trị, đặt nền móng cho các hệ thống an ninh mạng tự hành và kiên cường trong tương lai.