Tổng quan nghiên cứu

Trí tuệ nhân tạo đang tạo ra bước nhảy vọt trong nhiều ngành công nghiệp mũi nhọn như y tế, giao thông thông minh và an ninh giám sát. Trong đó, Mạng sinh đối nghịch (Generative Adversarial Networks - GAN) nổi lên như một công nghệ then chốt với khả năng tái tạo và tổng hợp dữ liệu hình ảnh vượt trội. Tuy nhiên, quá trình suy luận (inference) của mạng GAN đòi hỏi khối lượng tính toán khổng lồ với hàng triệu tham số phức tạp, khiến việc triển khai trên các thiết bị biên (edge devices) gặp trở ngại lớn về độ trễ và dung lượng bộ nhớ. Các vi xử lý thông thường như CPU đa nhân hay GPU tiêu chuẩn thường bộc lộ hạn chế về hiệu suất năng lượng và tính linh hoạt khi vận hành tại các nút cảm biến biên.

Trước thách thức này, nghiên cứu tập trung giải quyết bài toán tăng tốc phần cứng cho mạng Generator trong GAN bằng cách thiết kế kiến trúc chuyên dụng trên nền tảng SoC-FPGA. Mục tiêu cụ thể là xây dựng bộ tăng tốc Deconvolution đa nhân (DCMKP) với 4 nhân xử lý song song, tối ưu hóa đường truyền dữ liệu và áp dụng kỹ thuật lượng tử hóa dấu phẩy tĩnh 8-bit. Nghiên cứu được triển khai thực nghiệm toàn diện trên kit phát triển Xilinx Zynq UltraScale+ MPSoC ZCU106 và Kria KV260 tại Trường Đại học Bách khoa - ĐHQG TP.HCM vào tháng 06/2024. Kết quả đạt được mang ý nghĩa đột phá khi nâng cao tốc độ thực thi lên 58 lần so với vi xử lý máy tính tiêu chuẩn, đồng thời cải thiện hiệu quả năng lượng hơn 400 lần, mở ra hướng đi khả thi cho các ứng dụng thị giác máy tính thời gian thực.

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 trên nền tảng lý thuyết mạng nơ-ron sâu và cấu trúc đối nghịch của GAN do Ian Goodfellow khởi xướng. Mô hình bao gồm mạng Generator có nhiệm vụ biến đổi vector nhiễu ngẫu nhiên thành dữ liệu giả lập và mạng Discriminator phân định ranh giới giữa dữ liệu thực và dữ liệu nhân tạo thông qua hàm mất mát Binary Cross-Entropy. Trọng tâm toán học nằm ở lớp tích chập chuyển vị (Deconvolution hay Transpose Convolution), thuật toán chủ đạo để nâng cao độ phân giải không gian của đặc trưng ảnh. Mối quan hệ kích thước đầu ra được chuẩn hóa theo công thức toán học với các tham số kích thước đầu vào, bước nhảy stride và kích thước bộ lọc kernel.

Kiến trúc phần cứng tích hợp mô hình điện toán lượng tử hóa số thực dấu phẩy tĩnh 8-bit thay thế cho định dạng 32-bit floating-point truyền thống, kết hợp cơ chế kiểm soát tràn số chủ động. Toàn bộ hệ thống vận hành trên kiến trúc phân tầng SoC-FPGA, chia tách rõ ràng giữa Hệ thống xử lý (Processing System - PS) sử dụng lõi ARM Cortex và Logic lập trình (Programmable Logic - PL) chứa các khối tăng tốc phần cứng chuyên biệt.

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

Nguồn dữ liệu thực nghiệm bao gồm ba bộ dữ liệu tiêu chuẩn quốc tế: tập chữ số viết tay MNIST với 60.000 mẫu ảnh đơn sắc kích thước 28x28 pixel, tập khuôn mặt Celeb-A gồm 202.599 mẫu ảnh màu kích thước 64x64 pixel, và tập ảnh độ phân giải cao DIV2K gồm 1.000 mẫu dùng cho bài toán siêu phân giải hình ảnh. Cỡ mẫu nghiên cứu được trích xuất đại diện theo phương pháp chọn mẫu phân tầng ngẫu nhiên, đảm bảo độ đa dạng về góc nhìn và điều kiện ánh sáng để đánh giá toàn diện tính tổng quát của mô hình.

Phương pháp phân tích dựa trên mô phỏng chức năng và tổng hợp mạch số phần cứng thông qua bộ công cụ Xilinx Vivado Design Suite. Lý do lựa chọn phương pháp thiết kế mạch trực tiếp bằng ngôn ngữ Verilog HDL thay vì tổng hợp bậc cao (HLS) nhằm kiểm soát chặt chẽ từng chu kỳ xung nhịp, tối ưu hóa việc phân bổ tài nguyên DSP và BRAM on-chip. Hệ thống truyền nhận dữ liệu giữa PS và PL được thiết lập qua giao thức AXI4 kết hợp bộ điều khiển truy cập bộ nhớ trực tiếp trung tâm (CDMA). Quá trình đánh giá định lượng sử dụng các chỉ số học thuật khắt khe gồm điểm khoảng cách Fréchet Inception Distance (FID), tỷ số tín hiệu trên nhiễu đỉnh (PSNR) và chỉ số tương đồng cấu trúc (SSIM), song song với việc đo đạc công suất tiêu thụ thực tế bằng công cụ phân tích năng lượng chuyên sâu.

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

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

Thứ nhất, kiến trúc thực thi trực tiếp (Direct Implementation) cho thuật toán Deconvolution đã loại bỏ hoàn toàn 100% các phép toán nhân với giá trị 0 phát sinh từ kỹ thuật chèn zero-padding truyền thống. Giải pháp này giúp cắt giảm hơn 50% số chu kỳ tính toán lãng phí, tạo ra mẫu truy cập bộ nhớ đồng nhất trên chip FPGA.

Thứ hai, bộ xử lý Deconvolution đa nhân (DCMKP) với 4 lõi tính toán song song kết hợp cơ chế đệm Ping-Pong Buffer đã nâng cao thông lượng dữ liệu lên gấp 4 lần so với cấu trúc đơn nhân. Hệ thống duy trì xung nhịp hoạt động ổn định, đạt hiệu suất xử lý vượt trội trên tài nguyên 1.728 khối DSP của chip Zynq UltraScale+.

Thứ ba, kỹ thuật lượng tử hóa 8-bit fixed-point tích hợp cơ chế kẹp giá trị (Clamping) giúp giảm 75% băng thông truyền tải dữ liệu và dung lượng lưu trữ trên BRAM so với định dạng 32-bit float. Chất lượng ảnh tạo ra vẫn duy trì độ chính xác quang học cao với điểm số FID trên tập MNIST đạt mức 20.35, tương đương sai số dưới 2% so với mô hình huấn luyện bằng phần mềm gốc.

Thứ tư, hệ thống tăng tốc phần cứng đạt tốc độ xử lý nhanh hơn 58 lần so với CPU x86 tiêu chuẩn và vượt trội 3.6 lần so với các dòng GPU máy tính bàn thông thường, trong khi mức tiêu thụ điện năng giảm hơn 400 lần so với kiến trúc vi xử lý truyền thống.

Thảo luận kết quả

Hiệu năng đột phá của hệ thống bắt nguồn từ việc kết hợp đồng bộ giữa kiến trúc luồng dữ liệu đường ống (fully-pipelined) và phân vùng địa chỉ bộ nhớ liên tục trong không gian 32KB cho từng segment BRAM từ địa chỉ 0xE000_0000 đến 0xE003_FFFF. Việc chia sẻ không gian BRAM giữa bản đồ đặc trưng đầu vào và kết quả đầu ra đã triệt tiêu độ trễ chuyển giao dữ liệu ngoại vi.

Khi đối chiếu với các công trình nghiên cứu trước đây trong cùng lĩnh vực, giải pháp của nhóm tác giả khắc phục triệt để hiện tượng nghẽn cổ chai giao tiếp vốn xuất hiện trong phương pháp lặp ngược (reverse looping) của Zhang và cộng sự, đồng thời không gây suy giảm chất lượng biên ảnh như giải pháp phân khối của Li và cộng sự. Dữ liệu thực nghiệm được trực quan hóa qua biểu đồ tương quan giữa thông lượng (GOPs) và độ trễ suy luận, phản ánh đường cong tăng trưởng tuyến tính vượt trội. Bảng tổng hợp tài nguyên phần cứng khẳng định thiết kế chỉ chiếm dụng khoảng 65% số lượng LUT và Flip-Flop trên kit ZCU106, bảo đảm khả năng mở rộng linh hoạt cho các mạng nơ-ron phức tạp hơn.

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

Thứ nhất, tối ưu hóa việc phân bổ tài nguyên bộ nhớ nội vi bằng cách khai thác sâu 96 khối UltraRAM (URAM) sẵn có trên dòng chip UltraScale+ nhằm mục tiêu tăng dung lượng đệm trung gian lên 40% trong lộ trình 6 tháng, do các kỹ sư thiết kế phần cứng số chủ trì.

Thứ hai, mở rộng quy mô nhân tính toán từ cấu hình 4 nhân hiện tại lên 8 hoặc 16 nhân DCMKP song song đối với các dòng FPGA công nghiệp cỡ lớn, đặt mục tiêu nâng thông lượng xử lý chạm ngưỡng 150 GOPs trong vòng 12 tháng tới do nhóm nghiên cứu kiến trúc máy tính thực hiện.

Thứ ba, tích hợp toàn diện mô hình siêu phân giải SRGAN tăng tốc phần cứng vào hệ thống chẩn đoán hình ảnh y khoa thực tế (như nâng cao độ phân giải ảnh chụp cộng hưởng từ MRI) và camera giám sát thông minh, với mục tiêu rút ngắn thời gian xử lý xuống dưới 15 mili-giây mỗi khung hình trong giai đoạn 18 tháng do các đơn vị phát triển giải pháp thị giác biên triển khai.

Thứ tư, xây dựng gói công cụ phần mềm tự động hóa (Auto-deployment Framework) dựa trên nền tảng PYNQ, cho phép chuyển đổi trực tiếp các mô hình GAN từ PyTorch sang bitstream phần cứng trong thời gian 9 tháng do các kỹ sư phần mềm hệ thống nhúng đảm trách.

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

Nhóm kỹ sư thiết kế vi mạch và phần cứng số: Tiếp cận chi tiết phương pháp thiết kế bộ xử lý tích chập chuyển vị đa nhân bằng Verilog HDL, kỹ thuật quản lý pipeline và giải pháp giao tiếp tối ưu qua AXI4 CDMA.

Nhóm nghiên cứu khoa học dữ liệu và trí tuệ nhân tạo: Tham khảo quy trình lượng tử hóa mô hình sinh, kỹ thuật xử lý tràn số trong tính toán số thực dấu phẩy tĩnh và phương pháp đánh giá chất lượng ảnh sinh bằng chỉ số FID và SSIM.

Các doanh nghiệp phát triển thiết bị biên và IoT thông minh: Tìm kiếm giải pháp công nghệ nâng cao độ phân giải ảnh trực tiếp tại biên với mức tiêu thụ năng lượng cực thấp, ứng dụng cho robot tự hành và thiết bị bay không người lái.

Học viên cao học và sinh viên chuyên ngành Kỹ thuật Máy tính: Sử dụng làm tài liệu tham khảo chuẩn mực về phương pháp đồng thiết kế phần cứng - phần mềm (Hardware/Software Co-design) trên nền tảng Xilinx Zynq MPSoC.

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

Phương pháp Direct Implementation mang lại lợi thế kỹ thuật gì so với phương pháp Zero-Padding truyền thống? Phương pháp Direct Implementation áp dụng trực tiếp các bộ lọc lên dữ liệu đầu vào, giúp triệt tiêu 100% các phép tính nhân vô nghĩa với số 0 phát sinh từ việc chèn viền. Điều này giúp tối ưu hóa hơn 50% tài nguyên xử lý và đảm bảo tính đồng nhất của luồng dữ liệu trên FPGA.

Việc chuyển đổi sang định dạng 8-bit fixed-point có làm suy giảm chất lượng ảnh sinh ra không? Thực nghiệm cho thấy chất lượng ảnh sinh ra không bị suy giảm đáng kể nhờ cơ chế kẹp ngưỡng chống tràn số chủ động. Điểm số FID trên tập dữ liệu MNIST chỉ chênh lệch dưới 2% so với tính toán dấu phẩy động 32-bit trên máy tính, trong khi dung lượng bộ nhớ giảm tới 75%.

Nền tảng SoC-FPGA ZCU106 vượt trội hơn CPU và GPU ở những khía cạnh nào? SoC-FPGA kết hợp hoàn hảo giữa tính linh hoạt của phần mềm trên chip ARM và sức mạnh tính toán song song của phần cứng FPGA. Hệ thống đạt tốc độ xử lý cao hơn 58 lần so với CPU, đồng thời tiêu thụ năng lượng thấp hơn 400 lần so với kiến trúc máy tính truyền thống và hiệu quả hơn GPU chuyên dụng gấp nhiều lần.

Kiến trúc DCMKP giải quyết vấn đề chồng lấn điểm ảnh trong phép toán Deconvolution như thế nào? Kiến trúc DCMKP tích hợp bộ xử lý chồng lấn chuyên dụng (Core Overlap Processor) kết hợp mạch tích lũy hàng và cột theo cấu trúc đường ống. Cơ chế này tự động cộng dồn các giá trị pixel chồng lấn ngay trong chu kỳ xử lý mà không cần đọc ghi nhiều lần vào bộ nhớ ngoài.

Cơ chế truyền dữ liệu DMA qua giao thức AXI4 đóng vai trò gì trong hệ thống? Bộ điều khiển Central DMA (CDMA) cho phép truyền nhận trực tiếp các khối dữ liệu đặc trưng và trọng số giữa DRAM của vi xử lý và các khối BRAM trên FPGA với dung lượng 32KB mỗi khối. Quá trình này diễn ra tự động ở tốc độ cao mà không chiếm dụng chu kỳ lệnh của CPU.

Kết luận

Thứ nhất, luận văn đã thiết kế và hiện thực hóa thành công kiến trúc tăng tốc phần cứng chuyên biệt cho thuật toán Deconvolution trong mạng GAN trên nền tảng Xilinx Zynq UltraScale+ MPSoC.

Thứ hai, việc kết hợp phương pháp Direct Implementation và lượng tử hóa 8-bit đã loại bỏ hoàn toàn các phép toán thừa, giảm 75% dung lượng bộ nhớ mà vẫn duy trì chất lượng ảnh sinh chân thực.

Thứ ba, hệ thống đạt hiệu năng vượt trội với tốc độ xử lý tăng 58 lần so với CPU và hiệu quả năng lượng cao hơn 400 lần, khẳng định tính ưu việt của giải pháp phần cứng tùy biến.

Thứ tư, nghiên cứu đã chứng minh tính khả thi cao khi ứng dụng thành công vào bài toán siêu phân giải hình ảnh SRGAN trên các tập dữ liệu thực tế như MNIST, Celeb-A và DIV2K.

Thứ năm, công trình tạo tiền đề vững chắc cho việc thương mại hóa các thiết bị thị giác máy tính tiết kiệm năng lượng tại rìa mạng trong lộ trình phát triển 12 đến 24 tháng tới. Quý độc giả và các đơn vị nghiên cứu quan tâm có thể khai thác tài liệu này để triển khai các hệ thống trí tuệ nhân tạo nhúng hiệu năng cao trong thực tiễn.