Tổng quan nghiên cứu

Học đa tác vụ (Multi-Task Learning - MTL) đóng vai trò nền tảng trong sự phát triển của trí tuệ nhân tạo hiện đại, hiện diện trong hơn 85% các hệ thống thị giác máy tính, xử lý ngôn ngữ tự nhiên và học đa phương thức phức tạp. Khi một mô hình trí tuệ nhân tạo được huấn luyện đồng thời trên nhiều tác vụ có mối tương quan, mục tiêu cốt lõi là tận dụng các cấu trúc đặc trưng dùng chung để nâng cao hiệu năng dự báo của từng tác vụ riêng lẻ. Tuy nhiên, việc tối ưu hóa đồng thời các hàm mất mát thường xuyên đối mặt với hiện tượng chuyển tiếp âm (negative transfer), làm suy giảm từ 5% đến 15% độ chính xác tổng thể so với các mô hình đơn nhiệm biệt lập. Nguyên nhân then chốt dẫn đến sự suy giảm này là xung đột gradient, xảy ra khi các véc-tơ đạo hàm của các tác vụ tạo thành góc tù, khiến bước cập nhật tham số có lợi cho tác vụ này lại làm tổn hại trực tiếp đến tác vụ khác.

Luận văn Thạc sĩ Khoa học ngành Toán Tin mang tên "Nghiên cứu một số mô hình học sâu đa mục tiêu" được hoàn thành vào tháng 10/2023 tại Viện Toán ứng dụng và Tin học, Đại học Bách khoa Hà Nội dưới sự hướng dẫn của TS. Trần Ngọc Thăng. Mục tiêu của công trình là hệ thống hóa các giải thuật tối ưu hóa đa mục tiêu kinh điển, phân tích chuyên sâu các phương pháp điều chỉnh gradient hiện đại và đề xuất thuật toán mới mang tên AdaCAGrad (Adaptive Conflict-Averse Gradient Descent). Nghiên cứu giải quyết rào cản lớn của thuật toán CAGrad tiền nhiệm liên quan đến sự phụ thuộc vào hằng số Lipschitz, tạo ra cơ chế tự điều chỉnh tỉ lệ học tối ưu. Luận văn mở ra hướng tiếp cận toán học vững chắc, giúp cải thiện tốc độ hội tụ và nâng cao độ chính xác trên các bài toán tối ưu đa mục tiêu đa chiều trong khoảng thời gian nghiên cứu thực nghiệm từ đầu năm 2023 đến cuối năm 2023.

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

Khung lý thuyết áp dụng

Khung lý thuyết của luận văn được xây dựng trên nền tảng của Lý thuyết Tối ưu hóa Đa mục tiêu (Multi-Objective Optimization - MOO) và Giải tích lồi. Bài toán học đa mục tiêu (Multi-Objective Learning - MOL) được mô hình hóa dưới dạng tìm điểm cực tiểu của một hàm véc-tơ gồm $p$ hàm chi phí thành phần trên không gian tham số thực $n$ chiều $\mathbb{R}^n$. Khái niệm trọng tâm bao gồm:

  • Điểm tối ưu Pareto và Mặt Pareto (Pareto front): Điểm tham số mà tại đó không thể cải thiện giá trị của bất kỳ hàm mục tiêu nào nếu không làm tổn hại đến ít nhất một hàm mục tiêu khác.
  • Điểm dừng Pareto (Pareto-stationary point): Điều kiện cần cấp một cho tối ưu Pareto, biểu diễn qua sự tồn tại của véc-tơ trọng số $\lambda$ thuộc đơn hình xác suất $p$ chiều sao cho tổ hợp lồi của các gradient thành phần bằng 0.
  • Hiện tượng xung đột gradient: Được xác định theo tiêu chuẩn tích vô hướng $\langle g_i, g_j \rangle < 0$, tương ứng góc $\phi_{ij}$ giữa gradient của hai tác vụ $i$ và $j$ là góc tù ($\cos \phi_{ij} < 0$).

Bên cạnh hai phương pháp kinh điển là Hướng giảm dốc nhất đa mục tiêu (MSDM) và Thuật toán Newton đa mục tiêu (MNM), luận văn phân tích 4 thuật toán gradient hiện đại: Thuật toán hướng giảm đa gradient (MGDA), Phương pháp phẫu thuật gradient (PCGrad), Gradient Sign Dropout (GradDrop) và Hướng giảm chống xung đột (CAGrad). Trên cơ sở đó, thuật toán đề xuất AdaCAGrad tích hợp bài toán đối ngẫu tìm hướng cập nhật chống xung đột với nguyên lý tự thích nghi bước nhảy, khắc phục hoàn toàn điểm nghẽn ước lượng hằng số Lipschitz $H$.

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

Nghiên cứu áp dụng phương pháp giải tích toán học kết hợp mô phỏng thực nghiệm trên máy tính nhằm kiểm chứng tính đúng đắn và hiệu năng thuật toán. Nguồn dữ liệu thực nghiệm gồm 2 phần:

  • Bài toán tối ưu hóa tổng quát 2 chiều mô phỏng hình học không gian trạng thái với 3 điểm khởi tạo khác nhau nhằm khảo sát quỹ đạo hội tụ.
  • Bộ dữ liệu thị giác máy tính Multi-Fashion+MNIST gồm 70.000 mẫu ảnh kết hợp từ 2 tập dữ liệu chuẩn mực: 60.000 ảnh huấn luyện và 10.000 ảnh kiểm thử. Phương pháp chọn mẫu áp dụng kỹ thuật ghép cặp ảnh phân tầng ngẫu nhiên giữa tập chữ số viết tay MNIST (kích thước $28 \times 28$ pixel, đặt ở góc trên bên trái) và tập sản phẩm thời trang FashionMNIST ($28 \times 28$ pixel, đặt ở góc dưới bên phải), tạo thành bức ảnh tổng hợp kích thước $36 \times 36$ pixel phục vụ cho việc giải quyết 2 tác vụ phân loại song song.

Lý do lựa chọn kiến trúc mạng nơ-ron Multi-LeNet với 50 epochs huấn luyện, kích thước batch 256 và bộ tối ưu Adam là để đảm bảo sự đồng nhất trong so sánh thực nghiệm với các công trình chuẩn quốc tế. Tỉ lệ học khởi tạo được thiết lập tại mức $\alpha_0 = 0.05$, các tham số điều khiển thuật toán gồm siêu tham số chống xung đột $c = 0.4$, hệ số suy giảm bước nhảy $\kappa = 0.5$ và hệ số kiểm định Armijo $\sigma = 0.5$.

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

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

Quá trình đánh giá thực nghiệm định lượng trên các tập thử nghiệm chuẩn mực mang lại 3 phát hiện khoa học mang tính đột phá:

  • Nâng cao độ chính xác dự báo toàn diện: Trên tập dữ liệu kiểm thử Multi-Fashion+MNIST, thuật toán đề xuất AdaCAGrad đạt độ chính xác 96.58% đối với Tác vụ 1 (nhận dạng chữ số) và 88.91% đối với Tác vụ 2 (nhận dạng vật phẩm thời trang). Kết quả này vượt trội hơn tất cả các phương pháp đối chuẩn hàng đầu: vượt MGDA (95.80% và 86.95%), vượt PCGrad (96.05% và 87.82%), vượt GradDrop (95.90% và 87.50%), vượt CAGrad nguyên bản (96.25% và 88.10%) và vượt cả phương pháp huấn luyện tác vụ đơn độc lập (96.12% và 88.40%).
  • Tối ưu hóa hàm chi phí trung bình: Giá trị hàm mất mát trung bình của mô hình khi huấn luyện bằng AdaCAGrad giảm xuống mức 0.38, đạt mức thấp nhất so với 0.46 của MGDA, 0.43 của PCGrad, 0.44 của GradDrop và 0.40 của CAGrad, chứng minh khả năng tìm kiếm nghiệm tiệm cận tối ưu toàn cục tốt hơn khoảng 5% đến 17%.
  • Triệt tiêu hiện tượng dao động khi bước nhảy lớn: Trong bài toán mô phỏng 2 chiều, khi tỉ lệ học khởi tạo tăng từ $\alpha = 0.05$ lên $\alpha = 0.8$, các thuật toán như Gradient Descent thông thường và CAGrad bị dao động mạnh quanh mặt Pareto, kéo dài số bước lặp. Ngược lại, cơ chế tự thích nghi của AdaCAGrad tự động co bước nhảy theo điều kiện giảm thực tế, tạo ra đường cong quỹ đạo mượt mà và giảm số vòng lặp hội tụ xuống khoảng 40%.

Thảo luận kết quả

Thành công của thuật toán AdaCAGrad bắt nguồn từ sự kết hợp chặt chẽ giữa việc tìm kiếm hướng gradient trong không gian đối ngẫu thấp chiều $\mathbb{R}^p$ và cơ chế kiểm soát bước nhảy tự động. Trong khi các giải thuật truyền thống như MGDA thường bị dừng sớm tại các điểm dừng Pareto ngẫu nhiên có chất lượng kém, và PCGrad chỉ loại bỏ xung đột trực giao mà không kiểm soát được điểm hội tụ mục tiêu, AdaCAGrad trực tiếp tối ưu hóa tốc độ giảm của hàm chi phí trung bình.

Dữ liệu thực nghiệm được trực quan hóa qua hệ thống bảng biểu so sánh chi tiết và đồ thị quỹ đạo tối ưu 2 chiều. Đồ thị tiến trình hàm mất mát và sự biến thiên của tỉ lệ học theo từng epoch chứng minh rằng tỉ lệ học $\alpha_t$ chỉ thay đổi một số hữu hạn lần trong các giai đoạn đầu, sau đó ổn định ở một ngưỡng tối ưu. Điều này khẳng định tính đúng đắn của định lý giới hạn dãy bước nhảy được chứng minh trong phần lý thuyết giải tích của luận văn.

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

Dựa trên các đóng góp toán học và kết quả thực nghiệm đạt được, luận văn đưa ra 4 nhóm đề xuất hành động cụ thể:

  • Tích hợp thuật toán AdaCAGrad vào các hệ thống AI công nghiệp: Các kỹ sư Machine Learning và viện nghiên cứu trí tuệ nhân tạo nên áp dụng AdaCAGrad vào các kiến trúc xử lý ảnh y tế, xe tự hành và mô hình ngôn ngữ lớn đa nhiệm. Mục tiêu định lượng là tăng độ chính xác phân loại từ 1.5% đến 2.5% và rút ngắn 20% thời gian hội tụ GPU, triển khai áp dụng trong giai đoạn 3 đến 6 tháng tới.
  • Chuẩn hóa quy trình tiền xử lý và thiết kế hàm mất mát đa nhiệm: Các nhóm kỹ thuật dữ liệu cần thiết lập ngưỡng kiểm soát góc xung đột giữa các gradient tác vụ ngay trong giai đoạn thiết kế kiến trúc mạng, duy trì góc phân kỳ trung bình dưới 60 độ, hoàn thành chuẩn hóa quy trình trong Quý II/2024.
  • Xây dựng gói thư viện mã nguồn mở chuyên dụng: Nhóm nghiên cứu thuộc các trường đại học công nghệ cần phát triển các module tích hợp sẵn cho PyTorch và TensorFlow dựa trên mã nguồn của luận văn. Mục tiêu đạt trên 500 lượt trích dẫn khoa học và phục vụ giảng dạy cao học trong vòng 12 tháng tới.
  • Tối ưu hóa tính toán cho bài toán quy mô lớn: Đối với các bài toán có từ 10 tác vụ trở lên, các nhà khoa học dữ liệu cần nghiên cứu kết hợp kỹ thuật lấy mẫu tập con tác vụ ngẫu nhiên nhằm giảm độ phức tạp tính toán thời gian từ cấp lũy thừa xuống mức tuyến tính, thực hiện trong kế hoạch nghiên cứu giai đoạn 2024 - 2025.

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

Công trình luận văn thạc sĩ này là nguồn tài liệu học thuật và kỹ thuật giá trị cao cho các nhóm đối tượng sau:

  • Giảng viên, Nghiên cứu sinh và Học viên cao học ngành Toán Tin và Khoa học Dữ liệu: Sử dụng làm tài liệu tham khảo chuyên sâu về giải tích lồi, tối ưu hóa đa mục tiêu, các điều kiện KKT và kỹ thuật chứng minh sự hội tụ của thuật toán tối ưu phi tuyến.
  • Kỹ sư Trí tuệ Nhân tạo và Machine Learning: Ứng dụng thuật toán vào việc giải quyết bài toán xung đột gradient trong các hệ thống gợi ý thương mại điện tử, thị giác máy tính và nhận dạng tiếng nói đa tác vụ.
  • Sinh viên đại học năm cuối khối ngành Công nghệ Thông tin và Toán Ứng dụng: Tham khảo phương pháp xây dựng mô hình nghiên cứu, kỹ thuật thiết kế mạng nơ-ron sâu Multi-LeNet và quy trình thực nghiệm khoa học chuẩn mực.
  • Trưởng nhóm R&D và Kiến trúc sư giải pháp AI: Định hình chiến lược tối ưu hóa hạ tầng điện toán, giảm thiểu chi phí phần cứng huấn luyện mô hình phức tạp thông qua việc cải tiến giải thuật tối ưu.

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

Hiện tượng xung đột gradient trong học sâu đa mục tiêu là gì?
Xung đột gradient xảy ra khi tích vô hướng giữa các véc-tơ gradient của hai tác vụ mang giá trị âm, tức góc giữa chúng vượt quá 90 độ. Khi đó, việc cập nhật trọng số theo hướng giảm mất mát của tác vụ này sẽ trực tiếp làm gia tăng hàm mất mát của tác vụ kia, gây ra hiện tượng chuyển tiếp âm làm giảm hiệu năng chung.

Thuật toán AdaCAGrad có điểm gì vượt trội so với thuật toán CAGrad nguyên bản?
CAGrad đòi hỏi người dùng phải chọn tỉ lệ học thủ công thỏa mãn điều kiện phụ thuộc vào hằng số Lipschitz của gradient, vốn rất khó ước lượng chính xác trên mạng nơ-ron sâu. AdaCAGrad tích hợp cơ chế tự thích nghi, tự động kiểm tra bất đẳng thức suy giảm để điều chỉnh bước nhảy mượt mà mà không cần biết trước hằng số Lipschitz.

Vì sao không nên chỉ sử dụng phương pháp gán trọng số cố định cho các hàm chi phí?
Phương pháp tổng trọng số cố định không thể điều chỉnh được sự chênh lệch về tốc độ học và độ lớn gradient giữa các tác vụ trong quá trình huấn luyện. Điều này khiến tác vụ có gradient lớn chiếm ưu thế hoàn toàn, làm suy giảm nghiêm trọng độ chính xác của các tác vụ còn lại.

Hiệu năng thực tế của AdaCAGrad trên tập dữ liệu Multi-Fashion+MNIST như thế nào?
Thực nghiệm cho thấy AdaCAGrad đạt độ chính xác 96.58% trên tác vụ chữ số và 88.91% trên tác vụ thời trang, đồng thời kéo giảm hàm chi phí trung bình xuống 0.38, vượt trội hơn tất cả các phương pháp nổi tiếng như MGDA, PCGrad, GradDrop và CAGrad.

Thuật toán AdaCAGrad có làm gia tăng đáng kể chi phí tính toán không?
Không. Bằng cách chuyển đổi bài toán tối ưu gradient từ không gian tham số hàng triệu chiều của mạng nơ-ron sang không gian đơn hình trọng số đối ngẫu có số chiều bằng số lượng tác vụ $p$, thuật toán giải quyết bài toán con cực tiểu với chi phí thời gian tính toán thực nghiệm gần như không đáng kể.

Kết luận

  • Luận văn hệ thống hóa toàn diện cơ sở toán học của các giải thuật tối ưu đa mục tiêu từ kinh điển (MSDM, MNM) đến các phương pháp gradient tiên tiến (MGDA, PCGrad, GradDrop, CAGrad).
  • Đề xuất thành công thuật toán đột phá AdaCAGrad, kết hợp hài hòa giữa nguyên lý chống xung đột gradient và cơ chế tự điều chỉnh tỉ lệ học thích nghi theo điều kiện giảm hàm mục tiêu.
  • Chứng minh chặt chẽ về mặt toán học tính hội tụ của dãy bước nhảy và nghiệm tiệm cận Pareto mà không phụ thuộc vào hằng số Lipschitz định trước.
  • Kiểm chứng thực nghiệm xuất sắc trên tập dữ liệu ảnh Multi-Fashion+MNIST với độ chính xác đạt 96.58% và 88.91%, thiết lập kỷ lục hiệu năng mới so với các thuật toán đối chuẩn.
  • Mở rộng triển vọng ứng dụng thuật toán vào các mô hình học sâu quy mô lớn phục vụ sản xuất thực tế trong giai đoạn 2024 - 2025.

Các nhà nghiên cứu, kỹ sư học máy và bạn đọc quan tâm có thể tiếp cận toàn văn luận văn, tham khảo khung lý thuyết giải tích và khai thác mã nguồn thuật toán để áp dụng trực tiếp vào các dự án tối ưu hóa trí tuệ nhân tạo chuyên sâu.