Tổng quan nghiên cứu

Sự phát triển mạnh mẽ của trí tuệ nhân tạo và học sâu đã tạo ra bước đột phá lớn trong các bài toán thị giác máy tính và xử lý ngôn ngữ tự nhiên. Tuy nhiên, các mô hình mạng nơ-ron học sâu ngày càng mở rộng về quy mô, đòi hỏi năng lực tính toán khổng lồ với hàng triệu tham số. Trong thực tế, các phép tính tích chập chiếm hơn 80% tổng khối lượng tính toán của một mạng nơ-ron tích chập tiêu chuẩn. Việc triển khai các mô hình phức tạp này lên các thiết bị điện toán biên đang đối mặt với rào cản nghiêm trọng do giới hạn ngặt nghèo về tài nguyên phần cứng và công suất tiêu thụ năng lượng.

Nghiên cứu tập trung giải quyết bài toán nghẽn cổ chai tính toán trên thiết bị biên thông qua việc thiết kế và hiện thực một hệ thống tăng tốc mạng nơ-ron học sâu chuyên dụng trên nền tảng phần cứng tái cấu hình FPGA SoC. Mục tiêu cụ thể là xây dựng kiến trúc lõi tính toán song song hóa cao độ, tối ưu hóa luồng dữ liệu và tái sử dụng bộ nhớ cục bộ để giảm thiểu truy cập bộ nhớ ngoài. Phạm vi nghiên cứu được triển khai thực nghiệm từ tháng 9 năm 2020 đến tháng 6 năm 2021 tại Trường Đại học Bách Khoa thuộc Đại học Quốc gia Thành phố Hồ Chí Minh, sử dụng kit phát triển Avnet Ultra96-v2 tích hợp chip Xilinx Zynq UltraScale+ MPSoC ZU3EG.

Ý nghĩa khoa học và thực tiễn của công trình thể hiện qua việc tạo ra một lõi IP tăng tốc độc lập, hoạt động ở tần số 300 MHz, đạt tốc độ xử lý nhanh hơn 15.7 lần so với bộ vi xử lý ARM Cortex-A53 4 nhân chạy ở xung nhịp 1.2 GHz. Kết quả này chứng minh tiềm năng vượt trội của giải pháp phần cứng tùy biến trong việc tối ưu hóa năng lượng và nâng cao hiệu suất xử lý thời gian thực cho các thiết bị thông minh thế hệ mới.

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 xây dựng trên nền tảng lý thuyết mạng nơ-ron nhân tạo đa tầng và mạng nơ-ron tích chập chuyên sâu. Cấu trúc mạng tích chập tiêu chuẩn bao gồm ba tầng cơ bản: tầng tích chập thực hiện rút trích đặc trưng không gian qua toán tử nhân tích lũy ma trận, tầng gộp giúp giảm chiều dữ liệu và tầng kết nối đầy đủ phục vụ phân loại đầu ra. Trong miền rời rạc, phép tích chập hai chiều đa kênh kết hợp các tham số đệm và sải bước đòi hỏi hàng triệu phép tính nhân và cộng dồn liên tiếp.

Để giải quyết bài toán tiêu hao năng lượng do truy xuất dữ liệu liên tục từ bộ nhớ ngoài, luận văn áp dụng khung lý thuyết tối ưu hóa kiến trúc cố định trọng số. Hai chiến lược cốt lõi được khai thác triệt để gồm:

  • Tái sử dụng dữ liệu theo thời gian: Giữ nguyên giá trị trọng số trong các thanh ghi nội vi suốt nhiều chu kỳ tính toán, rút ngắn khoảng cách tái sử dụng xuống mức tối thiểu bằng 1 chu kỳ để hạn chế đọc bộ nhớ.
  • Tái sử dụng dữ liệu theo không gian: Phân phối đồng thời một phần tử dữ liệu đầu vào đến nhiều khối xử lý trên cùng một chu kỳ xung nhịp để thực hiện song song hóa phép tính.

Các khái niệm kỹ thuật trọng tâm bao gồm đơn vị xử lý phần tử, tổng riêng phần, bộ xử lý đa bộ lọc - kênh, bộ đệm toàn cục và giao thức truyền dữ liệu AXI.

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

Nghiên cứu áp dụng phương pháp thiết kế hệ thống từ trên xuống, phân tách rõ ràng giữa lớp phần mềm điều khiển và lớp phần cứng tính toán. Toàn bộ kiến trúc lõi tăng tốc được mô tả bằng ngôn ngữ đặc tả phần cứng Verilog HDL có khả năng tổng hợp cao, đảm bảo tính độc lập và tương thích đa nền tảng.

Dữ liệu kiểm thử sử dụng tập hợp các ma trận hình ảnh và trọng số đa chiều với kích thước linh hoạt từ 1xNxNxC đến KxMxMxC theo định dạng số nguyên 8-bit cố định. Phương pháp chọn mẫu dữ liệu được thực hiện theo nguyên tắc quét toàn diện các cấu hình tích chập phổ biến trong thực tế, bao gồm kích thước bộ lọc 3x3, các mức đệm từ P=0 đến P=1 và sải bước từ S=1 đến S=3. Việc lựa chọn phương pháp phân tích thực nghiệm so sánh trực tiếp giúp đánh giá khách quan độ trễ, thông lượng và hiệu quả sử dụng tài nguyên giữa phần cứng tăng tốc với các bộ xử lý đa nhân.

Hệ thống được mô phỏng chức năng logic trên công cụ Xilinx Vivado Design Suite và nạp bitstream kiểm chứng trực tiếp trên phần cứng thực nghiệm Avnet Ultra96-v2 trong giai đoạn 9 tháng nghiên cứu. Hệ thống tích hợp bộ vi xử lý ARM Cortex-A53 4 nhân, bộ nhớ ngoài 2 GB LPDDR4, khối điều khiển truyền dữ liệu trực tiếp AXI CDMA cùng 784 KB bộ nhớ Block RAM trên chip chia thành 3 bộ đệm chuyên biệt.

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

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

Quá trình tổng hợp và thực nghiệm trên phần cứng đã mang lại các phát hiện kỹ thuật mang tính đột phá:

  • Khả năng xử lý song song vượt trội: Lõi tính toán tích hợp 3 bộ xử lý đa bộ lọc - kênh, mỗi bộ chứa ma trận 3x4 khối xử lý phần tử, cho phép thực hiện đồng thời 36 phép tính nhân tích lũy trên 3 kênh đầu vào, 4 bộ lọc và 3 vị trí trượt trong mỗi chu kỳ xung nhịp 300 MHz.
  • Tốc độ tăng tốc ấn tượng: Phần cứng tăng tốc đạt tốc độ xử lý phép tích chập hai chiều nhanh 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.4 lần so với bộ xử lý máy tính để bàn Intel 6 nhân khi chạy thuần phần mềm.
  • Tối ưu hóa truyền nhận dữ liệu: Việc ứng dụng khối AXI CDMA chuyên dụng giúp giải phóng 100% thời gian của CPU chủ trong khâu vận chuyển dữ liệu ma trận giữa bộ nhớ ngoài và bộ nhớ trong chip.
  • Tiết kiệm tài nguyên phần cứng: Toàn bộ thiết kế chiếm dụng tỷ lệ tài nguyên rất khiêm tốn trên chip Zynq ZU3EG, sử dụng dưới 20% trong tổng số 70,560 bảng tra cứu và tận dụng hiệu quả 216 khối Block RAM nội vi, mở ra không gian lớn để tích hợp thêm các module xử lý khác.

Thảo luận kết quả

Hiệu năng vượt trội của hệ thống xuất phát từ việc giải quyết thành công điểm nghẽn truy xuất bộ nhớ thông qua cơ chế cố định trọng số. Khi một ma trận trọng số được nạp vào bộ đệm nội vi 16 KB, các giá trị này được giữ cố định trong thanh ghi cục bộ của từng phần tử xử lý để phục vụ liên tiếp cho nhiều luồng dữ liệu ảnh đầu vào, loại bỏ hoàn toàn độ trễ đọc dữ liệu từ thanh RAM ngoài 2 GB LPDDR4.

Khi quan sát dữ liệu biểu diễn qua biểu đồ cột so sánh thời gian thực thi, thời gian xử lý của phần cứng tăng tốc duy trì ở mức thấp và tuyến tính khi số lượng bộ lọc tăng từ 4 lên 32 bộ lọc. Ngược lại, đồ thị thời gian thực thi của bộ xử lý ARM 4 nhân và Intel 6 nhân tăng vọt theo hàm số mũ do hiện tượng trễ dòng lệnh và hạn chế băng thông bộ nhớ đệm cache.

Bảng số liệu kiểm thử chi tiết cũng khẳng định khi thay đổi kích thước ma trận đầu vào từ 14x14 đến 112x112 với sải bước S=1 hoặc S=3, phần cứng luôn duy trì hệ số tăng tốc ổn định trên 15 lần. Kết quả này hoàn toàn tương thích với các phân tích lý thuyết và vượt trội hơn so với một số nghiên cứu tăng tốc sử dụng phương pháp tính tuần tự trên FPGA trước đây.

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

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

  • Mở rộng quy mô mảng phần tử xử lý: Tăng số lượng bộ xử lý đa bộ lọc - kênh từ 3 lên 6 khối và mở rộng ma trận phần tử xử lý lên 4x8 nhằm nâng tổng năng lực tính toán lên 192 phép nhân tích lũy song song trên mỗi chu kỳ xung nhịp. Mục tiêu đạt hiệu năng xử lý trên 100 GOPS vào quý 4 năm 2026, do nhóm kỹ sư thiết kế phần cứng vi mạch chủ trì thực hiện.
  • Tích hợp kỹ thuật lượng tử hóa thích ứng và nén thưa: Áp dụng phương pháp lượng tử hóa dữ liệu 4-bit kết hợp loại bỏ các trọng số có giá trị tiệm cận 0 nhằm giảm 50% dung lượng bộ đệm Block RAM và tăng 30% tốc độ thực thi. Kế hoạch triển khai trong vòng 6 tháng tới dưới sự phối hợp giữa chuyên gia thuật toán học máy và kỹ sư FPGA.
  • Xây dựng trình biên dịch tự động hóa luồng lệnh: Phát triển bộ công cụ phần mềm cho phép tự động chuyển đổi các mô hình học sâu chuẩn như MobileNet hoặc ResNet thành tập lệnh cấu hình thanh ghi điều khiển cho lõi tăng tốc. Thời gian hoàn thiện trong 3 tháng, do đội ngũ phát triển phần mềm hệ thống nhúng đảm nhận.
  • Áp dụng kỹ thuật quản lý năng lượng động: Thiết lập cơ chế ngắt xung nhịp cục bộ đối với các khối xử lý tạm thời nhàn rỗi trong quá trình tính toán đệm biên, giúp tiết kiệm 25% công suất tiêu thụ toàn hệ thống trên các thiết bị IoT dùng pin. Thời gian kiểm định trong 12 tháng bởi bộ phận kỹ thuật phần cứng.

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

Luận văn mang lại giá trị học thuật và ứng dụng thực tiễn cao cho bốn nhóm đối tượng chuyên môn:

  • Kỹ sư thiết kế phần cứng vi mạch và FPGA: Nắm bắt phương pháp luận thiết kế hệ thống trên chip, kỹ thuật xây dựng lõi IP xử lý phép tích chập đa kênh và giải pháp quản lý bộ nhớ đệm BRAM tốc độ cao trong các dự án phát triển chip AI thương mại.
  • Nhà nghiên cứu và lập trình viên Trí tuệ nhân tạo biên: Tiếp cận phương pháp tối ưu hóa toán tử học sâu từ góc độ kiến trúc phần cứng, hiểu rõ cơ chế tái sử dụng dữ liệu không gian và thời gian để thiết kế các mô hình mạng nơ-ron thân thiện với thiết bị nhúng.
  • Giảng viên và học viên cao học chuyên ngành Khoa học Máy tính: Sử dụng công trình làm tài liệu tham khảo chất lượng cao cho các học phần về Kiến trúc máy tính nâng cao, Thiết kế hệ thống nhúng và Đồng thiết kế phần cứng - phần mềm trên nền tảng Xilinx Zynq.
  • Doanh nghiệp phát triển giải pháp IoT và Camera thông minh: Vận dụng kiến trúc gia tốc phần cứng để xây dựng các sản phẩm camera AI nhận diện khuôn mặt và vật thể thời gian thực, giúp cắt giảm 40% chi phí phần cứng và tối ưu hóa điện năng vận hành.

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

  • Tại sao nền tảng FPGA lại vượt trội hơn CPU và GPU trong các ứng dụng AI ở biên? FPGA cho phép tùy biến mạch logic trực tiếp để xử lý song song các phép tính ma trận mà không cần qua tầng biên dịch lệnh phức tạp như CPU. Với xung nhịp hoạt động 300 MHz, FPGA tiêu thụ công suất dưới 5W nhưng vẫn đạt tốc độ xử lý nhanh hơn 15.7 lần so với CPU nhúng, trong khi GPU đòi hỏi nguồn điện lớn vượt quá giới hạn thiết bị biên.

  • Kiến trúc cố định trọng số mang lại lợi thế gì cho hệ thống tăng tốc? Cơ chế này lưu giữ các trọng số trong thanh ghi cục bộ của 12 phần tử xử lý suốt quá trình trượt qua các điểm ảnh đầu vào. Giải pháp này giúp cắt giảm hơn 70% số lần truy xuất bộ nhớ ngoài DRAM 2 GB, triệt tiêu độ trễ nghẽn băng thông và giảm thiểu điện năng tiêu thụ toàn mạch.

  • Hệ thống có khả năng tương thích với các kiến trúc mạng nơ-ron khác nhau không? Có. Hệ thống được thiết kế dưới dạng lõi IP độc lập điều khiển thông qua bảng thanh ghi cấu hình. Người dùng có thể linh hoạt thiết lập các thông số về kích thước ảnh đầu vào, số kênh, số bộ lọc, độ rộng bước nhảy và kích thước đệm để chạy nhiều mô hình học sâu như AlexNet, VGG hoặc MobileNet.

  • Khối AXI CDMA đóng vai trò gì trong việc tối ưu hóa hiệu năng? Khối Central Direct Memory Access hoạt động như một kênh truyền dữ liệu tốc độ cao độc lập giữa bộ nhớ ngoài và 784 KB bộ nhớ Block RAM trên chip. Cơ chế này giải phóng 100% thời gian xử lý của 4 nhân CPU ARM Cortex-A53, cho phép vi xử lý tập trung vào các tác vụ điều khiển ứng dụng cấp cao.

  • Thiết kế có thể chuyển dịch sang các dòng chip FPGA khác được không? Hoàn toàn khả thi. Toàn bộ mã nguồn lõi tính toán được viết bằng ngôn ngữ chuẩn Verilog HDL và thiết kế ở mức trừu tượng hóa cao. Thiết kế có thể dễ dàng tổng hợp lại trên các dòng chip khác như Xilinx Artix-7, Kintex-7 hay các dòng FPGA của Intel mà không cần thay đổi cấu trúc vi kiến trúc lõi.

Kết luận

  • Công trình đã 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 phần cứng tái cấu hình FPGA SoC.
  • Kiến trúc đề xuất đạt tốc độ thực thi phép tích chập hai chiều nhanh gấp 15.7 lần so với bộ xử lý ARM 4 nhân và vượt qua bộ xử lý Intel 6 nhân ở tần số hoạt động 300 MHz.
  • Thiết kế kết hợp hiệu quả 36 đơn vị nhân tích lũy song song với cơ chế cố định trọng số, tối ưu hóa triệt để tài nguyên bộ nhớ đệm Block RAM nội vi.
  • Toàn bộ mục tiêu nghiên cứu đã được hoàn thành xuất sắc trong thời gian 9 tháng thực nghiệm trên kit phát triển Avnet Ultra96-v2.
  • Luận văn cung cấp một giải pháp phần cứng mở, độc lập và khả thi cao cho các ứng dụng trí tuệ nhân tạo thời gian thực trên thiết bị điện toán biên.

Trong giai đoạn 12 tháng tiếp theo, nhóm nghiên cứu hướng tới việc tích hợp thuật toán lượng tử hóa 4-bit và mở rộng mảng tính toán lên 192 đơn vị xử lý. Hãy ứng dụng ngay giải pháp kiến trúc phần cứng tiên tiến này để nâng tầm hiệu năng cho các hệ thống nhúng thông minh của bạn.