Giới thiệu dự án

Sự bùng nổ của mạng lưới vạn vật kết nối Internet (IoT - Internet of Things) cùng các ứng dụng trí tuệ nhân tạo (AI) đã thúc đẩy nhu cầu xử lý dữ liệu thông minh tại các thiết bị biên (Edge Devices). Theo các báo cáo công nghiệp gần đây, hàng tỷ thiết bị IoT đang tạo ra hàng zettabyte dữ liệu mỗi năm. Trong mô hình học máy truyền thống (Centralized Machine Learning), toàn bộ dữ liệu thô từ các nút biên phải được truyền tải và tập trung về máy chủ trung tâm để huấn luyện. Tuy nhiên, cách tiếp cận này đối mặt với các rủi ro nghiêm trọng về nghẽn băng thông mạng, độ trễ truyền thông và đặc biệt là nguy cơ vi phạm quyền riêng tư theo các khung pháp lý quốc tế nghiêm ngặt như Quy định chung về bảo vệ dữ liệu (GDPR - General Data Protection Regulation).

Để giải quyết mâu thuẫn giữa khai thác dữ liệu và bảo mật thông tin, học tập liên kết (Federated Learning - FL) được giới thiệu như một mô hình điện toán biên phân tán tiên tiến. Thay vì chia sẻ dữ liệu cục bộ, các thiết bị tham gia (Clients) chỉ huấn luyện mô hình trên dữ liệu tại chỗ và gửi các tham số cập nhật (trọng số, độ dốc gradient) về máy chủ điều phối (Central Server) để tổng hợp thành mô hình toàn cục thông qua các thuật toán như Federated Averaging (FedAvg).

Mặc dù giải quyết được bài toán bảo vệ quyền riêng tư dữ liệu thô, kiến trúc phân tán của học tập liên kết lại bộc lộ những lỗ hổng an ninh mạng mới. Do máy chủ trung tâm hoàn toàn không thể trực tiếp kiểm tra dữ liệu hoặc quá trình huấn luyện nội bộ của client (nhất là khi tích hợp các giao thức mã hóa tổng hợp an toàn - Secure Aggregation), các tác nhân độc hại có thể lợi dụng quyền kiểm soát thiết bị biên để thực hiện tấn công đầu độc (Poisoning Attack).

Đồ án tốt nghiệp chuyên ngành Công nghệ Kỹ thuật Máy tính với đề tài "Nghiên cứu phương pháp tấn công dựa trên mạng đối nghịch tạo sinh nhắm vào học liên kết trong điện toán biên cho hệ thống IoT" tập trung giải quyết bài toán: Làm thế nào một nút biên bị xâm nhập có thể thực hiện tấn công đầu độc dữ liệu (đảo nhãn - label flipping) đạt hiệu quả cao mà không cần biết trước tập dữ liệu huấn luyện của các nút biên khác trong hệ thống FL IoT?

Dự án xác định các mục tiêu nghiên cứu cụ thể:

  1. Nghiên cứu cơ sở lý thuyết về học tập liên kết, mạng đối nghịch tạo sinh (Generative Adversarial Networks - GAN) và các cơ chế tấn công đầu độc trong hệ thống phân tán.
  2. Xây dựng thuật toán tạo dữ liệu độc hại Data_Gen và mô hình tấn công PoisonGAN, cho phép kẻ tấn công tái tạo mẫu dữ liệu của các client lành tính thông qua tham số mô hình toàn cục.
  3. Tích hợp cơ chế khuếch đại gradient với hệ số tỷ lệ $S$ nhằm tối ưu hóa khả năng duy trì ảnh hưởng độc hại sau quá trình tổng hợp FedAvg.
  4. Triển khai mô hình thực nghiệm hoàn chỉnh trên nền tảng phần mềm mã nguồn mở Flower Framework và đo kiểm trực tiếp trên phần cứng nhúng Raspberry Pi 4 Model B.
  5. Đánh giá định lượng hiệu quả tấn công thông qua độ chính xác nhiệm vụ chính (Main Accuracy), độ chính xác đầu độc (Poison Accuracy) và chi phí tài nguyên phần cứng (CPU, RAM, thời gian thực thi).

Phạm vi nghiên cứu tập trung vào kịch bản tấn công đảo nhãn trên bộ dữ liệu tiêu chuẩn MNIST (chuyển nhãn lớp chữ số "2" thành lớp chữ số "7"), môi trường thực nghiệm gồm 33 clients với 1 attacker, sử dụng mạng nơ-ron tích chập (CNN) triển khai trên bo mạch nhúng Raspberry Pi 4 (4GB RAM) kết nối mạng cục bộ.

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

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

Các kỹ thuật tấn công đầu độc trong học tập liên kết hiện nay chủ yếu tập trung vào tấn công mô hình cục bộ hoặc tấn công cửa sau (Backdoor Attacks). Tuy nhiên, hầu hết các phương pháp truyền thống đều đòi hỏi kẻ tấn công phải sở hữu một lượng tri thức nền tảng nhất định (prior knowledge) như một phần tập dữ liệu kiểm thử hoặc phân phối dữ liệu của các client khác.

Tiêu chí so sánh Alternating Minimization Attack Model Placement Backdoor Byzantine-robust Poisoning PoisonGAN (Giải pháp đề xuất)
Kiến thức yêu cầu Cần ước tính cập nhật của client lành tính Cần can thiệp cấu trúc mô hình toàn cục Cần tập dữ liệu xác thực (Validation set) Chỉ cần tham số mô hình toàn cục ($G_t$)
Cơ chế tạo dữ liệu Tạo nhiễu ngẫu nhiên thủ công Chèn dấu hiệu nhận biết (Trigger pattern) Tối ưu hóa gradient ngược (Back-gradient) Sinh dữ liệu tự động bằng GAN đối kháng
Thời điểm tấn công Phải duy trì liên tục qua nhiều vòng Kích hoạt khi có điều kiện trigger Duy trì liên tục qua các vòng giao tiếp Tấn công hiệu quả ngay trong 1 vòng (Single-round)
Khả năng tàng hình Trung bình Thấp (Dễ bị phát hiện qua trigger) Trung bình (Dễ bị bộ lọc Byzantine loại bỏ) Cao (Tận dụng phân phối dữ liệu non-IID)
Thực thi trên IoT Chưa tối ưu trên thiết bị biên Yêu cầu tài nguyên xử lý phức tạp Tính toán gradient tốn kém Đã kiểm chứng trên Raspberry Pi 4

Yêu cầu hệ thống được phân loại theo mô hình MoSCoW:

  • Must have (Bắt buộc): Tái tạo mẫu dữ liệu mục tiêu bằng GAN dựa trên mô hình toàn cục; thực hiện đảo nhãn lớp định trước; duy trì độ chính xác của các lớp không bị tấn công; tích hợp hoàn chỉnh với Flower Framework.
  • Should have (Nên có): Cơ chế khuếch đại tham số $S$ chống suy giảm gradient; khả năng chạy ổn định trên phần cứng giới hạn tài nguyên như Raspberry Pi 4.
  • Could have (Có thể có): Khả năng tùy biến linh hoạt số lượng client tham gia ngẫu nhiên trong mỗi vòng giao tiếp; hệ thống giám sát tiến trình phần cứng tự động qua psutil.
  • Won't have (Chưa thực hiện): Tấn công đồng thời nhiều lớp mục tiêu phức tạp trên các tập dữ liệu ảnh màu độ phân giải cao; vượt qua các giao thức phòng thủ mã hóa đa bên (SMPC) cấp độ phần cứng.

Thách thức kỹ thuật lớn nhất nằm ở sự giới hạn năng lực tính toán của các nút biên IoT. Thiết bị nhúng như Raspberry Pi 4 không có GPU chuyên dụng, do đó việc huấn luyện đồng thời mạng Generator ($G$) và Discriminator ($D$) đòi hỏi tối ưu hóa cấu trúc mạng nơ-ron nhằm tránh tràn bộ nhớ RAM và quá nhiệt CPU.

Thiết kế hệ thống

Kiến trúc hệ thống bao gồm máy chủ điều phối trung tâm và cụm nút mạng biên (Clients). Kẻ tấn công tham gia vào hệ thống như một client thông thường, tuân thủ giao thức bắt tay của Flower Framework nhưng thực thi tiến trình sinh dữ liệu độc và khuếch đại cập nhật mô hình trước khi truyền tải.

sequenceDiagram
    autonumber
    participant Server as Central Server (Flower FedAvg)
    participant Victim as Benign Clients (Raspberry Pi 4)
    participant Attacker as Malicious Client (PoisonGAN)
    
    Server->>Victim: 1. Phát tán tham số toàn cục G_t
    Server->>Attacker: 1. Phát tán tham số toàn cục G_t
    Note over Victim: Huấn luyện mô hình cục bộ L_t trên tập dữ liệu sạch D_real
    Note over Attacker: Sao chép G_t vào Discriminator D
    Note over Attacker: Generator G nhận nhiễu z -> Tạo mẫu giả x_fake
    Note over Attacker: Lọc mẫu x_fake (Lớp mục tiêu) -> Gán nhãn sai -> Tạo D_poison
    Note over Attacker: Huấn luyện mô hình cục bộ trên D_poison -> Thu được L^p
    Note over Attacker: Khuếch đại cập nhật: L^{p*} = S * L^p
    Victim->>Server: 2. Gửi cập nhật cục bộ lành tính L_{t+1}
    Attacker->>Server: 2. Gửi cập nhật độc hại L^{p*}_{t+1}
    Note over Server: Tổng hợp trọng số toàn cục: G_{t+1} = FedAvg(L_{t+1}, L^{p*}_{t+1})

Công nghệ sử dụng trong hệ thống:

  • Ngôn ngữ lập trình: Python 3.9+
  • Khung học sâu: PyTorch 1.12+ (hỗ trợ tính toán gradient và xây dựng các lớp tích chập)
  • Khung học tập liên kết: Flower Framework (flwr 1.4+)
  • Hệ điều hành nút biên: Raspberry Pi OS (Debian 64-bit)
  • Công cụ đo kiểm tài nguyên: Thư viện psutil 5.9+

Chi tiết cấu trúc mạng nơ-ron được thiết kế tối ưu cho bài toán phân loại ảnh MNIST (kích thước $28 \times 28 \times 1$):

[Kiến trúc Mô hình Toàn cục / Discriminator]
Input (28x28x1 / 64x64x1)

[Kiến trúc Mô hình Generator]
Input (Nhiễu ngẫu nhiên z in R^100)
Output (Mẫu ảnh giả sinh ra: 64x64x1)

Methodology

Nghiên cứu áp dụng phương pháp thực nghiệm kết hợp giữa mô phỏng phần mềm và kiểm chứng phần cứng thực tế (Hardware-in-the-loop validation):

  1. Giai đoạn mô phỏng (Simulation Phase): Tiến hành thiết lập thuật toán, tối ưu siêu tham số trên nền tảng đám mây Google Colab và Kaggle (sử dụng GPU Nvidia T4) để kiểm tra tính hội tụ của hàm loss GAN và tốc độ lật nhãn.
  2. Giai đoạn nhúng thực tế (Hardware Deployment Phase): Chuyển giao mã nguồn sang cụm bo mạch Raspberry Pi 4 kết nối qua giao thức gRPC nội bộ của Flower.

Kế hoạch triển khai và quản trị rủi ro:

  • Rủi ro tràn bộ nhớ (OOM) trên Raspberry Pi: Khắc phục bằng cách cấu hình kích thước batch nhỏ (Batch size = 32 hoặc 64) và tối ưu hóa bộ thu gom rác bộ nhớ của Python sau mỗi epoch.
  • Rủi ro biến mất gradient (Vanishing Gradient) trong GAN: Thay thế hàm mục tiêu cực tiểu hóa $\min \log(1 - D(G(z)))$ bằng hàm cực đại hóa $\max \log(D(G(z)))$ trong giai đoạn đầu huấn luyện.

Implementation và kết quả

Development process

Cốt lõi của giải pháp là hai thuật toán: Thuật toán sinh dữ liệu độc hại Data_Gen và Thuật toán tấn công tổng thể PoisonGAN.

# Cốt lõi logic sinh dữ liệu độc hại Data_Gen
import torch
import torch.nn as nn

def data_gen_step(generator, discriminator, target_class, chosen_label, num_samples, noise_dim=100, device="cpu"):
    generator.eval()
    discriminator.eval()
    poison_samples = []
    
    criterion = nn.CrossEntropyLoss()
    
    while len(poison_samples) < num_samples:
        z = torch.randn(64, noise_dim, device=device)
        fake_images = generator(z)
        outputs = discriminator(fake_images)
        predictions = torch.argmax(outputs, dim=1)
        
        # Lọc các mẫu giả mạo được Discriminator nhận diện là lớp mục tiêu
        for idx in range(len(predictions)):
            if predictions[idx].item() == target_class:
                # Gán nhãn sai (đảo nhãn) do kẻ tấn công chỉ định
                poison_samples.append((fake_images[idx].detach(), chosen_label))
                if len(poison_samples) >= num_samples:
                    break
                    
    return poison_samples

Trong quá trình huấn luyện Generator, hàm mất mát được biểu diễn toán học như sau:

$$\mathcal{L}G(\theta_g) = \mathbb{E}{z \sim p(Z_{\text{noise}})} [\log(D(G(z)))]$$

Khi triển khai trên client độc hại, quy trình thực hiện theo các bước:

  1. Nhận trọng số $G_t$ từ Server, sao chép trực tiếp vào mạng Discriminator $D$.
  2. Đưa vector nhiễu ngẫu nhiên $z \sim \mathcal{N}(0, I)$ qua Generator $G$ để tạo mẫu giả $x_{\text{fake}} = G(z)$.
  3. Đưa $x_{\text{fake}}$ qua $D$. Nếu $D(x_{\text{fake}}) = y_m$ (lớp mục tiêu, ví dụ chữ số '2'), mẫu sẽ được lưu vào tập dữ liệu độc $D_{\text{poison}}$ với nhãn mới $y_n$ (lớp đảo nhãn, ví dụ chữ số '7').
  4. Huấn luyện mô hình cục bộ $L_t$ trên tập $D_{\text{poison}}$ để thu được cập nhật $L_{t+1}^p$.
  5. Mở rộng độ dốc cập nhật trước khi gửi về máy chủ bằng hệ số khuếch đại $S$:

$$L_{t+1}^{p*} = S \times L_{t+1}^p$$

Hệ số $S$ đóng vai trò quyết định giúp cập nhật độc hại không bị triệt tiêu khi máy chủ thực hiện tính trung bình cộng FedAvg với các client lành tính:

$$G_{t+1} = \sum_{k=1}^{K} \frac{n_k}{N} L_{t+1}^k$$

Testing và validation

Kịch bản thử nghiệm được cấu hình chi tiết:

  • Tập dữ liệu: MNIST gồm 70,000 ảnh ($28 \times 28$ pixel), chia thành 60,000 ảnh huấn luyện và 10,000 ảnh kiểm tra. Phân phối mẫu đồng đều giữa 33 clients (khoảng 1,818 mẫu/client).
  • Quy mô hệ thống: 33 clients, mỗi vòng chọn ngẫu nhiên 10 clients tham gia huấn luyện.
  • Client lành tính: Huấn luyện 10 epochs/vòng, Learning rate khởi tạo $\eta = 0.05$.
  • Client độc hại (Attacker): Huấn luyện 20 epochs/vòng, Learning rate $\eta = 0.05$, giảm 10 lần sau mỗi 2 epochs. Kẻ tấn công bắt đầu phát động tấn công sau khi mô hình toàn cục đạt độ chính xác ổn định từ 90%.
  • Tổng số vòng giao tiếp (Communication Rounds): 100 vòng.
[Chất lượng tái tạo mẫu chữ số '2' qua các vòng của Generator]
Round 20 : [ Khối nhiễu mờ, cấu trúc chưa định hình ]
Round 40 : [ Xuất hiện đường cong mờ của nét chữ số '2' ]
Round 80 : [ Đường viền chữ số '2' sắc nét, tương đương mẫu thật từ tập MNIST ]

Khảo sát tiêu hao tài nguyên phần cứng trung bình trong một vòng huấn luyện:

Thiết bị & Vai trò CPU Usage trung bình RAM Usage trung bình Max RAM tiêu thụ GPU Usage Thời gian thực thi / vòng
Raspberry Pi 4 - Victim 385% / 4 Cores 24% (0.96 GB / 4 GB) 32% (1.28 GB) N/A 5,784 giây (~1.61 giờ)
Raspberry Pi 4 - Attacker 385% / 4 Cores 32% (1.28 GB / 4 GB) 38% (1.52 GB) N/A 22,503 giây (~6.25 giờ)
Kaggle T4 - Victim 106% / 4 vCPUs 9.31% (2.70 GB / 29 GB) 11.2% (3.25 GB) 45% 17 giây
Kaggle T4 - Attacker 110% / 4 vCPUs 14.37% (4.17 GB / 29 GB) 16.5% (4.78 GB) 53% 73 giây

Dung lượng tệp tham số mô hình sau mỗi vòng đóng gói đạt xấp xỉ 4.18 MB, hoàn toàn phù hợp để truyền tải qua các chuẩn mạng không dây IoT (Wi-Fi, 4G/LTE).

Kết quả đạt được

Độ chính xác (%)
       Round 20   Round 50   Round 80
  1. Hiệu suất sinh dữ liệu: Tại vòng giao tiếp thứ 80, Generator $G$ đã học được đầy đủ phân phối xác suất của lớp chữ số "2" từ trọng số toàn cục, tạo ra các mẫu giả mạo có độ tương đồng cao với ảnh thực tế trong tập MNIST mà không cần truy cập trực tiếp dữ liệu của client khác.
  2. Hiệu quả tấn công đầu độc:
    • Poison Accuracy: Tăng vọt lên mức tiệm cận tuyệt đối trong các vòng có mặt client tấn công. Gần như toàn bộ các ảnh mang nhãn số "2" trong tập kiểm tra bị mô hình toàn cục nhận diện sai thành số "7".
    • Main Accuracy: Duy trì ổn định trên 85% đối với tất cả các lớp chữ số còn lại ($0, 1, 3, 4, 5, 6, 7, 8, 9$). Điều này chứng minh cuộc tấn công có tính chọn lọc cao, không phá hủy toàn bộ mô hình, khiến quản trị viên hệ thống rất khó phát hiện thông qua các chỉ số kiểm thử chung.
  3. Chi phí tính toán trên thiết bị biên: Client tấn công tiêu tốn thời gian thực thi gấp xấp xỉ 3.89 lần và bộ nhớ RAM cao hơn 33.3% so với client lành tính trên Raspberry Pi 4 do phải thực hiện vòng lặp sinh dữ liệu đối kháng và lọc nhãn.

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

  1. Cơ chế tấn công không phụ thuộc dữ liệu ngoài (Zero Prior-Data Dependency): Khác biệt hoàn toàn với các nghiên cứu trước đây đòi hỏi kẻ tấn công phải sở hữu một phần tập dữ liệu xác thực, mô hình PoisonGAN tận dụng trực tiếp các tham số mô hình toàn cục $G_t$ làm bộ phân biệt $D$. Đây là bước đột phá chứng minh rằng quyền riêng tư tham số trong FL vẫn có thể bị khai thác để đảo ngược dữ liệu.
  2. Chiến lược tấn công đơn vòng hiệu quả cao (Single-Round Impact): Nhờ tích hợp hệ số khuếch đại gradient $S$, PoisonGAN có khả năng bẻ cong hướng hội tụ của mô hình toàn cục ngay trong một vòng giao tiếp duy nhất. Cơ chế này vô hiệu hóa hoàn toàn các thuật toán phát hiện bất thường dựa trên phân tích chuỗi thời gian hành vi của client.
  3. Thực nghiệm hoàn chỉnh trên phần cứng nhúng thực tế: Đồ án là một trong những công trình tiên phong chuyển giao toàn bộ quy trình tấn công và huấn luyện học tập liên kết từ môi trường mô phỏng máy chủ đám mây sang mạng lưới thiết bị biên thực tế Raspberry Pi 4 Model B qua Flower Framework, cung cấp các số liệu đo kiểm tài nguyên (CPU, RAM, Execution Time) thực tế cho cộng đồng nghiên cứu AIoT.

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

Kịch bản ứng dụng và rủi ro thực tế

Mô hình tấn công nghiên cứu trong đồ án phản ánh các mối đe dọa an ninh mạng hiện hữu trong nhiều hệ thống thực tế:

  • Hệ thống y tế thông minh (Smart Healthcare): Nhiều bệnh viện cùng tham gia huấn luyện mô hình chẩn đoán hình ảnh X-quang/MRI bằng Federated Learning. Một nút biên bị nhiễm mã độc có thể dùng PoisonGAN để làm sai lệch kết quả chẩn đoán của một bệnh lý hiểm nghèo cụ thể trong khi các chẩn đoán khác vẫn hoạt động bình thường.
  • Giao thông thông minh và Xe tự hành (Connected Autonomous Vehicles): Các phương tiện chia sẻ thông tin nhận diện biển báo giao thông. Kẻ tấn công có thể làm hệ thống nhận diện biển báo "Dừng lại" (Stop) thành biển báo "Được phép chạy", gây tai nạn giao thông nghiêm trọng.
  • Giám sát lưới điện thông minh (Smart Grid IoT): Đầu độc các chỉ số đo lường điện năng từ các công tơ thông minh biên, dẫn đến sai lệch trong điều phối nguồn điện lưới khu vực.

Hướng dẫn triển khai thực nghiệm

# 1. Cài đặt môi trường trên Raspberry Pi 4 (Raspberry Pi OS 64-bit)
sudo apt-get update && sudo apt-get install -y python3-pip python3-dev libatlas-base-dev
pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
pip3 install flwr==1.4.0 psutil matplotlib

# 2. Khởi chạy Server điều phối trung tâm (Chạy trên máy tính điều khiển / Master Node)
python3 server.py --rounds 100 --min_clients 10

# 3. Khởi chạy các Client lành tính trên các bo mạch Raspberry Pi
python3 client_victim.py --server_address 192.168.1.100:8080 --client_id 1

# 4. Khởi chạy Client tấn công (PoisonGAN Node)
python3 client_attacker.py --server_address 192.168.1.100:8080 --target_class 2 --poison_label 7 --scale_factor 10

Lộ trình triển khai khuyến nghị cho hệ thống phòng thủ:

  • Giai đoạn 1 (Tháng 1-2): Đánh giá lỗ hổng hệ thống FL hiện tại bằng cách tích hợp module kiểm thử PoisonGAN.
  • Giai đoạn 2 (Tháng 3-4): Triển khai các thuật toán tổng hợp chống Byzantine (Krum, Trimmed Mean, Median) trên máy chủ điều phối Flower.
  • Giai đoạn 3 (Tháng 5-6): Tích hợp kỹ thuật quyền riêng tư vi sai (Differential Privacy - DP) bằng cách thêm nhiễu Gaussian vào gradient trước khi tổng hợp nhằm làm giảm độ sắc nét của dữ liệu do Generator tái tạo.

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

Hạn chế kỹ thuật

  • Thời gian xử lý trên CPU biên: Do Raspberry Pi 4 không tích hợp bộ tăng tốc AI (NPU/GPU), thời gian thực thi của client tấn công lên tới hơn 6 giờ cho một vòng huấn luyện đầy đủ 20 epochs, tạo ra độ trễ lớn trong mạng lưới thời gian thực.
  • Độ phức tạp của dữ liệu: Nghiên cứu mới dừng lại ở tập dữ liệu ảnh đơn sắc MNIST kích thước nhỏ ($28 \times 28$). Việc mở rộng sang ảnh màu độ phân giải cao (như CIFAR-10, ImageNet) đòi hỏi kiến trúc mạng GAN phức tạp hơn nhiều.
  • Giả định phân phối dữ liệu: Thử nghiệm hiện tại chia đều mẫu ngẫu nhiên (IID setting) trước khi phát động tấn công, chưa khảo sát sâu trên các phân phối dữ liệu thực tế bị lệch nghiêm trọng (Highly Non-IID).

Hướng phát triển tiếp theo

  • Khảo sát thuật toán PoisonGAN trên các dạng dữ liệu cảm biến chuỗi thời gian (Time-series) và dữ liệu bảng trong hệ thống điều khiển công nghiệp (SCADA IoT).
  • Nghiên cứu cơ chế tấn công thông đồng đa điểm (Multi-attacker collusion) với nhiều nút biên cùng phối hợp chia sẻ vector gradient độc hại.
  • Phát triển thuật toán phòng thủ mới dựa trên phân tích phổ vector gradient (Cosine Similarity Spectral Analysis) nhằm phát hiện và loại bỏ các bản cập nhật bị khuếch đại bởi hệ số $S$ mà không cần giải mã dữ liệu cá nhân.

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

   [ Sinh viên ]     [ Kỹ sư AIoT ]      [ Doanh nghiệp ]    [ Nhà nghiên cứu ]
   Tài liệu mẫu      Pattern thiết kế    Đánh giá bảo mật    Mở rộng thuật toán
   FL & GAN thực tế  mô hình trên Edge   hệ sinh thái FL     phòng thủ Byzantine
  • Sinh viên & Học viên chuyên ngành Máy tính / Điện tử: Nguồn tài liệu thực chứng toàn diện về cách kết hợp giữa Federated Learning, Generative Adversarial Networks và triển khai phần cứng nhúng thực tế thông qua Flower Framework.
  • Kỹ sư AIoT & Nhà phát triển hệ thống nhúng: Nắm bắt các phương thức tối ưu hóa bộ nhớ, quản lý tiến trình tính toán AI trên các dòng chip SoC giới hạn tài nguyên như Broadcom BCM2711.
  • Doanh nghiệp triển khai giải pháp Edge AI: Nhận thức rõ ràng các lỗ hổng bảo mật tiềm ẩn của học máy phân tán trước khi đưa vào vận hành sản xuất, từ đó chủ động xây dựng các lớp phòng thủ an ninh mạng.
  • Nhà nghiên cứu An toàn thông tin (Security Researchers): Cung cấp một phương pháp luận tấn công chuẩn mực (Benchmark baseline) để thử nghiệm và đánh giá độ bền vững của các thuật toán tổng hợp mới.

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

1. Yêu cầu cấu hình phần cứng tối thiểu để triển khai mô phỏng và thực nghiệm PoisonGAN?

Hệ thống yêu cầu tối thiểu:

  • Phía Client biên: Bo mạch Raspberry Pi 4 Model B (khuyến nghị phiên bản 4GB hoặc 8GB RAM), thẻ nhớ MicroSD Class 10 tối thiểu 32GB, nguồn điện ổn định 5V/3A.
  • Phía Server điều phối: Máy tính cá nhân chạy hệ điều hành Linux/Windows/macOS trang bị CPU 4 nhân, RAM 8GB (nếu chỉ điều phối mạng) hoặc máy trạm có GPU Nvidia (GTX 1650 / T4 trở lên) nếu muốn chạy mô phỏng tập trung quy mô lớn.

2. Giới hạn mở rộng (Scalability limits) của hệ thống FL trên thiết bị biên và giải pháp?

Giới hạn lớn nhất là độ trễ truyền thông và nghẽn băng thông khi số lượng thiết bị biên tăng lên hàng nghìn nút. Giải pháp là áp dụng chiến lược lựa chọn ngẫu nhiên một tập con client (ví dụ 10/33 clients như trong đồ án) trong mỗi vòng giao tiếp kết hợp kỹ thuật lượng tử hóa gradient (Gradient Quantization) để giảm dung lượng tệp truyền tải.

3. Làm thế nào để tích hợp thuật toán kiểm thử PoisonGAN vào các hệ thống FL hiện có?

Do PoisonGAN được xây dựng hoàn toàn tương thích với giao thức chuẩn của Flower Client (flwr.client.NumPyClient), người dùng chỉ cần kế thừa lớp client tiêu chuẩn và ghi đè (override) phương thức fit(), trong đó tích hợp module data_gen_step() và nhân hệ số tỷ lệ $S$ vào danh sách mảng NumPy của trọng số trước khi trả về cho Server.

4. Chi phí vận hành và nhu cầu bảo trì hệ thống học liên kết biên là gì?

Chi phí phần cứng cho mỗi nút biên IoT sử dụng Raspberry Pi 4 dao động khoảng 55 - 75 USD. Nhu cầu bảo trì chủ yếu tập trung vào việc duy trì kết nối mạng ổn định, quản lý tản nhiệt cho bo mạch khi chạy tải tính toán cao liên tục và cập nhật các bản vá bảo mật cho hệ điều hành Linux nhúng.

5. Tác động kinh tế và bài toán ROI khi đầu tư vào bảo mật học liên kết IoT?

Đầu tư vào việc đánh giá lỗ hổng bằng các công cụ như PoisonGAN giúp doanh nghiệp tránh được các vụ rò rỉ dữ liệu hoặc sai lệch quyết định tự động hóa có thể gây thiệt hại hàng triệu USD, đồng thời đảm bảo tuân thủ đầy đủ các tiêu chuẩn bảo mật dữ liệu quốc tế, rút ngắn thời gian đưa sản phẩm AIoT ra thị trường toàn cầu.

Kết luận

Đồ án tốt nghiệp "Nghiên cứu phương pháp tấn công dựa trên mạng đối nghịch tạo sinh nhắm vào học liên kết trong điện toán biên cho hệ thống IoT" đã thực hiện thành công việc xây dựng và chứng minh tính khả thi của mô hình tấn công đầu độc PoisonGAN kết hợp thuật toán sinh dữ liệu Data_Gen. Bằng cách biến đổi mô hình toàn cục thành bộ phân biệt Discriminator và ứng dụng cơ chế khuếch đại gradient $S$, kẻ tấn công có thể thao túng độ chính xác phân loại của lớp mục tiêu lên mức tuyệt đối mà vẫn giữ nguyên tính ổn định của toàn hệ thống (>85% Main Accuracy) mà không cần sở hữu tập dữ liệu thực tế nào từ trước.

Kết quả thực nghiệm trên phần cứng Raspberry Pi 4 qua khung làm việc Flower đã cung cấp bức tranh định lượng rõ nét về sự đánh đổi giữa năng lực tấn công và chi phí tài nguyên phần cứng trong môi trường điện toán biên thực tế. Đây là tiền đề học thuật và thực tiễn quan trọng, đóng góp nền tảng vững chắc cho việc phát triển các giải pháp phòng thủ tổng hợp an toàn thế hệ mới cho các hệ thống AIoT trong tương lai.