Tổng quan nghiên cứu

Sự bùng nổ của trí tuệ nhân tạo và mạng nơ-ron học sâu (Deep Neural Networks - DNN) đã tạo ra bước đột phá mạnh mẽ trong các lĩnh vực thị giác máy tính, xử lý ngôn ngữ tự nhiên và xe tự hành. Tuy nhiên, các mô hình học sâu hiện đại ngày càng trở nên phức tạp, đòi hỏi hàng tỷ phép tính mỗi giây cùng dung lượng bộ nhớ khổng lồ. Điều này tạo ra rào cản kỹ thuật nghiêm trọng khi triển khai trên các thiết bị điện toán biên (Edge Computing) – nơi tài nguyên phần cứng, băng thông và nguồn năng lượng luôn bị giới hạn nghiêm ngặt. Việc tính toán thuần túy bằng phần mềm trên các bộ vi xử lý nhúng ARM thông thường dẫn đến độ trễ cao và tiêu tốn nhiều điện năng.

Trước thách thức trên, đề tài luận văn thạc sĩ chuyên ngành Khoa học Máy tính của tác giả Nguyễn Xuân Quang, dưới sự hướng dẫn khoa học của PGS. Phạm Quốc Cường tại Trường Đại học Bách Khoa – ĐHQG TP.HCM (hoàn thành vào tháng 06 năm 2021), đã tập trung nghiên cứu và hiện thực thành công hệ thống tăng tốc mạng nơ-ron học sâu trên nền tảng phần cứng tái cấu hình FPGA SoC.

Mục tiêu trọng tâm của nghiên cứu là xây dựng kiến trúc phần cứng chuyên dụng có khả năng tối ưu hóa phép tích chập hai chiều (Conv2D) – phép toán chiếm hơn 80% khối lượng tính toán trong các mạng nơ-ron tích chập (CNN). Thiết kế được thử nghiệm thực tế trên kit phát triển Avnet Ultra96-v2 tích hợp chip Xilinx Zynq UltraScale+ MPSoC ZU3EG với bộ nhớ 2 GB LPDDR4. Kết quả thực nghiệm cho thấy hệ thống tăng tốc đạt xung nhịp 300 MHz, mang lại tốc độ thực thi vượt trội gấp 15.7 lần so với bộ xử lý ARM Cortex-A53 4 nhân và nhanh hơn khoảng 2.3 lần so với CPU máy tính để bàn Intel 6 nhân, mở ra hướng đi bền vững cho các ứng dụng trí tuệ nhân tạo ở biên.

Cơ sở lý thuyết và phương pháp nghiên cứu

Khung lý thuyết áp dụng

Nghiên cứu được xây dựng dựa trên nền tảng toán học của mạng nơ-ron nhân tạo và các kiến trúc học sâu tiêu biểu như AlexNet, VGG16 và MobileNet. Trọng tâm lý thuyết xoay quanh phép tích chập hai chiều đa kênh, trong đó dữ liệu đầu vào và các bộ trọng số (kernels) được xử lý thông qua chuỗi phép nhân và tích lũy (Multiply-Accumulate - MAC).

Để giải quyết nút thắt cổ chai về truy xuất bộ nhớ, luận văn ứng dụng hai nguyên lý tối ưu hóa phần cứng then chốt:

  1. Mô hình dòng dữ liệu cố định trọng số (Weight Stationary Dataflow): Giữ các giá trị trọng số trong thanh ghi cục bộ của từng đơn vị xử lý (Processing Element - PE) qua nhiều chu kỳ xung nhịp, giúp giảm tối đa số lần đọc từ bộ nhớ chính.
  2. Kỹ thuật tái sử dụng dữ liệu theo không gian (Spatial Data Reuse) và theo thời gian (Temporal Data Reuse): Cho phép phân phối đồng thời một phần tử dữ liệu đầu vào hoặc trọng số tới nhiều PE, rút ngắn khoảng cách tái sử dụng (reuse distance) xuống mức tối thiểu bằng 1 chu kỳ.

Mô hình nghiên cứu định hình cấu trúc tính toán số nguyên 8-bit (INT8) thông qua lượng tử hóa, kết hợp cơ chế định tuyến cộng dồn tổng thành phần (Partial Sum - psum) để đảm bảo độ chính xác của mạng mà vẫn tiết kiệm tài nguyên logic.

Phương pháp nghiên cứu

Nghiên cứu sử dụng phương pháp thực nghiệm kết hợp mô phỏng và kiểm thử trên phần cứng thực tế theo quy trình chuẩn từ ngày 21 tháng 09 năm 2020 đến ngày 13 tháng 06 năm 2021:

  • Nguồn dữ liệu và cỡ mẫu thực nghiệm: Bộ dữ liệu kiểm thử bao gồm các ma trận đặc trưng với kích thước đa dạng từ 8x8 đến 224x224 điểm ảnh, số lượng kênh từ 3 đến 64 kênh, kích thước bộ lọc 3x3 với các tham số đệm (padding P = 0 đến P = 1) và sải bước (stride S = 1 đến S = 3). Cỡ mẫu thực nghiệm gồm hơn 20 kịch bản tính toán tích chập đại diện cho các tầng xử lý thực tế trong các mô hình thị giác máy tính chuẩn.
  • Phương pháp chọn mẫu: Chọn mẫu theo cấu hình đặc trưng của các lớp tích chập phổ biến nhằm đánh giá đầy đủ tính ổn định của hệ thống khi kích thước đầu vào và số lượng bộ lọc thay đổi liên tục.
  • Phương pháp phân tích và hiện thực: Toàn bộ lõi tính toán được đặc tả bằng ngôn ngữ phần cứng Verilog HDL khả tổng hợp, thiết kế theo hướng trừu tượng cao, độc lập với nền tảng FPGA cụ thể. Tác giả sử dụng bộ công cụ Xilinx Vivado để phân tích tổng hợp tài nguyên (LUT, Flip-Flop, BRAM, DSP) và điều khiển luồng dữ liệu thông qua IP Core AXI CDMA kết nối bộ nhớ DRAM ngoài với bộ nhớ đệm BRAM trong chip. Hiệu năng được đo đạc chính xác qua thời gian thực thi (latency tính bằng mili-giây) và chỉ số tăng tốc (speedup ratio).

Kết quả nghiên cứu và thảo luận

Những phát hiện chính

Hệ thống tăng tốc phần cứng trên chip Xilinx Zynq UltraScale+ MPSoC đã mang lại những kết quả ấn tượng:

  • Nâng cao tốc độ xử lý vượt bậc: Phần cứng tăng tốc (HWAcc) hoạt động ở tần số 300 MHz đạt tốc độ thực thi nhanh hơn 15.7 lần so với bộ vi xử lý ARM Cortex-A53 4 nhân (chạy ở tần số 1.2 GHz) và nhanh hơn khoảng 2.3 lần so với vi xử lý Intel 6 nhân trên cùng các phép tính tích chập 2 chiều đa kênh.
  • Hiệu suất tính toán song song cao: Lõi tính toán tích hợp 3 bộ xử lý đa bộ lọc-kênh (KCPE), mỗi KCPE chứa ma trận 3x4 khối PE, cho phép thực thi đồng thời 36 phép tính MAC trên 3 kênh và 4 bộ lọc khác nhau trong từng chu kỳ xung nhịp.
  • Tối ưu hóa phân cấp bộ nhớ đệm on-chip: Hệ thống quản lý bộ nhớ gồm 256 KB Input Global Buffer, 16 KB Weight Global Buffer (chia thành 4 khối BRAM 4 KB độc lập) và 512 KB Output Global Buffer (chia thành 4 khối BRAM 128 KB) cho phép đọc ghi đồng thời đa luồng, loại bỏ hoàn toàn hiện tượng tranh chấp bộ nhớ.
  • Giảm độ trễ truyền dữ liệu: Nhờ sử dụng bộ điều khiển AXI Central DMA theo cơ chế memory-mapped, thời gian di chuyển dữ liệu giữa bộ nhớ ngoài LPDDR4 2 GB và các khối BRAM giảm hơn 80% so với phương pháp truyền dữ liệu truyền thống qua vi xử lý.

Thảo luận kết quả

Hiệu năng vượt trội của hệ thống xuất phát trực tiếp từ việc áp dụng thành công kiến trúc Weight Stationary kết hợp tối ưu luồng dữ liệu song song. Trong khi CPU truyền thống phải liên tục truy xuất dữ liệu từ bộ nhớ thứ cấp qua tập lệnh tuần tự, thiết kế FPGA cho phép các giá trị trọng số được lưu cố định tại các thanh ghi cục bộ của PE, giúp các phép tính nhân tích lũy được thực thi liên tục trên từng chu kỳ xung nhịp 300 MHz.

Khi so sánh với các công trình quốc tế, thiết kế này đạt sự cân bằng tối ưu giữa hiệu năng và tài nguyên. So với kiến trúc Winograd trên ZCU102 đạt 854.6 GOP/s cho mạng AlexNet hay bộ gia tốc neuFlow với 100 PE cố định, kiến trúc đề xuất có tính linh hoạt cao hơn nhờ khả năng tái cấu hình tham số lớp mạng thông qua khối Register File.

Trong các báo cáo khoa học, toàn bộ dữ liệu thực nghiệm được trực quan hóa sinh động qua biểu đồ đường so sánh thời gian thực thi giữa HWAcc, ARM Cortex-A53 và Intel CPU trên từng kích thước dữ liệu đầu vào. Đồng thời, bảng tổng hợp phân bổ tài nguyên từ Vivado minh chứng rằng thiết kế chỉ chiếm một phần hợp lý tài nguyên logic của chip ZU3EG (dưới 40% LUT và DSP), tạo dư địa lớn để tích hợp thêm các khối chức năng xử lý ảnh khác trên cùng một vi mạch.

Đề xuất và khuyến nghị

Dựa trên kết quả nghiên cứu thực nghiệm, tác giả đề xuất 4 giải pháp chiến lược nhằm hoàn thiện và mở rộng khả năng ứng dụng của hệ thống:

  • Nâng cấp lõi phần cứng hỗ trợ tích chập phân tách theo chiều sâu (Depthwise Separable Convolution) và tích chập giãn nở (Dilated Convolution): Tăng hiệu quả thực thi cho các mô hình mạng siêu nhẹ như MobileNetV3 hoặc EfficientNet thêm khoảng 35% trong vòng 6 tháng tới, thực hiện bởi nhóm nghiên cứu phần cứng nhúng.
  • Phát triển bộ công cụ phần mềm tự động biên dịch (Compiler Toolchain): Tự động lượng tử hóa mô hình từ framework PyTorch/TensorFlow và sinh trực tiếp các tập lệnh cấu hình thanh ghi trong vòng 9 tháng tới, do các kỹ sư phần mềm AI phụ trách.
  • Mở rộng kiến trúc bộ đệm kép (Ping-Pong Buffer) kết hợp AXI DMA đa luồng: Nâng thông lượng xử lý toàn hệ thống lên trên 50 GOP/s trong vòng 1 năm tới nhằm triệt tiêu hoàn toàn thời gian trễ khi nạp trọng số mới giữa các lớp mạng.
  • Triển khai thử nghiệm trên các hệ thống giám sát giao thông thông minh và thiết bị y tế cầm tay: Ứng dụng tại các đô thị lớn như TP. Hồ Chí Minh với mục tiêu giảm độ trễ nhận diện thời gian thực xuống dưới 30 mili-giây, thực hiện thông qua liên kết giữa trường đại học và các doanh nghiệp công nghệ cao trong giai đoạn 2022–2024.

Đối tượng nên tham khảo luận văn

Công trình luận văn là tài liệu tham khảo chuyên sâu có giá trị thực tiễn cao cho 4 nhóm đối tượng:

  • Kỹ sư thiết kế vi mạch và FPGA: Nắm bắt phương pháp đặc tả phần cứng bằng Verilog HDL, kỹ thuật thiết kế ma trận tính toán PE, kiến trúc bus AXI và bộ điều khiển CDMA trên các dòng chip hiện đại như Xilinx Zynq UltraScale+.
  • Giảng viên, nhà nghiên cứu và sinh viên ngành Khoa học Máy tính / Kỹ thuật Máy tính: Sử dụng làm tài liệu học thuật tiêu chuẩn về giải thuật tích chập, lý thuyết tái sử dụng dữ liệu không gian - thời gian và phương pháp đồng thiết kế phần cứng - phần mềm (Hardware/Software Co-design).
  • Kỹ sư phát triển hệ thống nhúng và IoT: Vận dụng các khối IP Core và driver phần mềm để xây dựng các thiết bị biên thông minh phục vụ nhận diện khuôn mặt, phân loại hình ảnh và giám sát an ninh cục bộ.
  • Lãnh đạo công nghệ và doanh nghiệp phát triển sản phẩm AI: Tham khảo căn cứ kỹ thuật để lựa chọn giải pháp phần cứng tối ưu chi phí, nâng cao hiệu quả năng lượng và rút ngắn thời gian đưa sản phẩm AIoT ra thị trường.

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

Tại sao nền tảng FPGA lại vượt trội hơn GPU trong các ứng dụng điện toán biên?

FPGA cho phép thiết kế phần cứng tùy biến sâu ở mức cổng logic, hoạt động ở tần số thấp khoảng 300 MHz nhưng đạt khả năng song song hóa cao. Nhờ đó, FPGA tiêu thụ năng lượng thấp hơn nhiều so với GPU, rất phù hợp với môi trường biên bị hạn chế điện năng.

Kiến trúc Weight Stationary mang lại lợi ích cụ thể gì cho hệ thống?

Kiến trúc này giữ cố định các trọng số trong thanh ghi cục bộ của từng khối PE trong suốt quá trình tính toán ma trận đầu vào. Cơ chế này giảm thiểu tối đa số lần truy cập bộ nhớ ngoài LPDDR4, cắt giảm hơn 80% độ trễ và tiết kiệm năng lượng đáng kể.

Hệ thống tăng tốc trong luận văn có thể xử lý đồng thời bao nhiêu phép tính?

Lõi tính toán gồm 3 bộ KCPE, mỗi bộ có 12 PE (ma trận 3x4). Nhờ đó, hệ thống thực hiện đồng thời 36 phép nhân tích lũy (MAC) 8-bit trên 3 kênh và 4 bộ lọc khác nhau trong mỗi chu kỳ xung nhịp 300 MHz.

Khối AXI CDMA đóng vai trò gì trong kiến trúc hệ thống?

AXI Central Direct Memory Access (CDMA) đóng vai trò trung chuyển khối dữ liệu lớn giữa bộ nhớ ngoài DRAM 2 GB và các khối BRAM nội vi theo cơ chế memory-mapped, giải phóng hoàn toàn bộ xử lý ARM khỏi tác vụ đọc ghi tuần tự chậm chạp.

Thiết kế này có thể mở rộng lên các dòng FPGA khác hoặc ASIC không?

Toàn bộ mã nguồn Verilog HDL của lõi tính toán được xây dựng độc lập với phần cứng cụ thể. Thiết kế hoàn toàn có thể tái cấu hình, nâng số lượng KCPE để tổng hợp trên các dòng FPGA quy mô lớn hơn hoặc chuyển đổi sang vi mạch chuyên dụng ASIC.

Kết luận

  • Đề tài đã thiết kế và hiện thực thành công hệ thống tăng tốc mạng nơ-ron học sâu hoàn chỉnh trên nền tảng FPGA SoC Xilinx Zynq UltraScale+ MPSoC.
  • Kiến trúc đề xuất đạt hiệu năng vượt trội, tăng tốc phép tích chập gấp 15.7 lần so với bộ xử lý ARM 4 nhân và 2.3 lần so với CPU Intel 6 nhân.
  • Khai thác hiệu quả kỹ thuật cố định trọng số và tái sử dụng dữ liệu, thực thi song song 36 phép MAC trên tần số hoạt động 300 MHz.
  • Tối ưu hóa toàn diện luồng dữ liệu thông qua hệ thống bộ nhớ phân cấp BRAM (tổng dung lượng 784 KB) kết hợp cơ chế truyền dữ liệu AXI CDMA tốc độ cao.
  • Đặt nền móng vững chắc cho lộ trình mở rộng hỗ trợ các kiến trúc mạng nơ-ron thế hệ mới trên thiết bị biên giai đoạn 2022–2024.

Quý độc giả, kỹ sư vi mạch và nhà nghiên cứu quan tâm có thể ứng dụng trực tiếp các giải pháp kiến trúc trong luận văn để nâng cao hiệu năng cho các hệ thống nhúng thông minh. Hãy kết nối và áp dụng ngay các kỹ thuật tăng tốc phần cứng tiên tiến này vào các dự án trí tuệ nhân tạo thực tiễn của bạn!