Tổng quan nghiên cứu

Trong bối cảnh dữ liệu hình ảnh và video ngày càng gia tăng với tốc độ nhanh chóng, việc xử lý và trích xuất thông tin quan trọng từ các bộ dữ liệu này trở thành một thách thức lớn. Theo ước tính, bộ dữ liệu NUS-WIDE chứa hơn 269.000 ảnh với hơn 5.000 thẻ khác nhau, trong khi một kho ảnh nội bộ quy mô lớn có tới 700.000 ảnh với hơn 24.000 khái niệm thẻ. Vấn đề đặt ra là làm thế nào để thu gọn danh sách các thẻ gán cho mỗi ảnh hoặc video mà vẫn giữ được khả năng đại diện chính xác cho nội dung. Bài toán Extractive Tags Summarization (ETS) được đề xuất nhằm giải quyết vấn đề này bằng cách tự động chọn ra các thẻ quan trọng từ bộ thẻ gốc, tránh việc tạo ra thẻ mới không có trong bộ thẻ ban đầu.

Mục tiêu nghiên cứu là phát triển một mô hình học sâu hiệu quả, có khả năng xử lý dữ liệu ảnh và mở rộng sang dữ liệu video, đồng thời giải quyết các vấn đề như mất cân bằng dữ liệu và sự xuất hiện của phần tử ngoại lai trong bộ thẻ. Phạm vi nghiên cứu tập trung vào dữ liệu ảnh và video thu thập từ các nguồn công khai và nội bộ, với thời gian nghiên cứu từ năm 2021 đến 2022. Ý nghĩa của nghiên cứu được thể hiện qua việc cải thiện độ chính xác phân loại thẻ, tăng tốc độ hội tụ mô hình và giảm thiểu ảnh hưởng của dữ liệu nhiễu, góp phần nâng cao hiệu quả tìm kiếm và quản lý dữ liệu đa phương tiệ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 dựa trên các lý thuyết và mô hình học sâu hiện đại, bao gồm:

  • Mạng nơ-ron nhân tạo (Artificial Neural Network - ANN): Cấu trúc gồm các lớp nơ-ron kết nối, sử dụng hàm kích hoạt phi tuyến để biểu diễn các quan hệ phức tạp trong dữ liệu.
  • Mạng nơ-ron tích chập (Convolutional Neural Network - CNN): Được sử dụng để trích xuất đặc trưng không gian từ ảnh, với các lớp tích chập và tổng hợp giúp giảm kích thước dữ liệu và tránh quá khớp.
  • Mạng nơ-ron Transformer: Sử dụng cơ chế Attention, đặc biệt là Multi-Head Attention và Self-Attention, để mô hình hóa mối quan hệ giữa các thẻ và đặc trưng ảnh, cho phép xử lý song song và hiệu quả hơn so với mạng hồi quy truyền thống.
  • Hệ hỗn hợp các chuyên gia (Mixture of Experts - MoE): Kết hợp nhiều mô hình con để tăng độ chính xác tổng thể, được áp dụng qua cơ chế Gating trong mô hình đề xuất.
  • Học tự giám sát (Self-supervised Learning): Phương pháp huấn luyện không cần nhãn dữ liệu, tận dụng dữ liệu chưa gán nhãn để tạo bài toán phụ, từ đó khởi tạo trọng số cho mô hình chính.

Các khái niệm chính bao gồm: hàm mất mát Binary Cross-Entropy (BCE), hàm mất mát Dice, hàm mất mát phụ trợ, cơ chế Attention (Queries, Keys, Values), và các kiến trúc mạng tích chập 2D, 3D, cũng như các biến thể như R3D, R(2+1)D, CSN, SlowFast, và Temporal Shift Module (TSM).

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

Nguồn dữ liệu chính gồm bộ dữ liệu công khai NUS-WIDE với 26.559 ảnh sau tiền xử lý và bộ dữ liệu nội bộ PIXTA Stock với 700.000 ảnh và 24.056 khái niệm thẻ. Dữ liệu video được thu thập từ kho video nội bộ với đặc điểm slow motion và các hành động lặp lại.

Phương pháp phân tích bao gồm:

  • Xây dựng mô hình học sâu đa luồng (MAGNeto) kết hợp luồng thẻ và luồng ảnh, sử dụng Transformer Encoder và Multi-Head Attention.
  • Áp dụng hàm mất mát kết hợp BCE-Dice để xử lý mất cân bằng dữ liệu.
  • Sử dụng hàm mất mát phụ trợ để cân bằng tín hiệu huấn luyện giữa các luồng.
  • Triển khai chiến lược học tự giám sát hai pha: huấn luyện trước với bài toán phát hiện phần tử ngoại lai và tinh chỉnh với học có giám sát.
  • Áp dụng kỹ thuật tăng cường dữ liệu Tag-adding và Tag-dropping để giảm ảnh hưởng phần tử ngoại lai.
  • Đối với dữ liệu video, nghiên cứu và lựa chọn mô hình trích xuất đặc trưng phù hợp như R3D, R(2+1)D, CSN, SlowFast, và TSM.
  • Chiến lược lựa chọn khung hình khác nhau cho giai đoạn huấn luyện (lấy mẫu ngẫu nhiên) và kiểm định (Center Crop, lấy mẫu đoạn video đại diện).

Quá trình nghiên cứu được thực hiện trên GPU RTX-2080ti với batch-size 32 cho NUS-WIDE và 256 cho bộ dữ liệu nội bộ, sử dụng thuật toán tối ưu SGD với momentum 0.9 và learning rate từ 10^-2 đến 3×10^-2 tùy bộ dữ liệu.

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

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

  1. Hiệu quả mô hình MAGNeto: Mô hình đề xuất MAGNeto đạt giá trị F1 khoảng 0.92, vượt trội so với các mô hình cơ sở TF-t (chỉ dùng thẻ) và TF-it (kết hợp thẻ và ảnh). Việc sử dụng hàm mất mát BCE-Dice giúp tăng tốc độ hội tụ và cải thiện độ chính xác, với F1 tăng từ khoảng 0.90 lên 0.92.

  2. Tác động của hàm mất mát phụ trợ: Việc bổ sung hàm mất mát phụ trợ giúp cân bằng tín hiệu huấn luyện giữa luồng thẻ và luồng ảnh, tránh hiện tượng luồng thẻ hội tụ quá nhanh làm luồng ảnh bị bỏ qua. Kết quả cho thấy mô hình sử dụng hàm mất mát phụ trợ hội tụ ổn định hơn và đạt F1 cao hơn khoảng 1-2%.

  3. Giảm ảnh hưởng phần tử ngoại lai: Kỹ thuật Tag-adding và Tag-dropping giúp mô hình giảm tỷ lệ ảnh bị ảnh hưởng bởi phần tử ngoại lai từ khoảng 30% xuống còn khoảng 12%, đồng thời duy trì hoặc cải thiện F1 trên bộ dữ liệu NUS-WIDE.

  4. Chiến lược học tự giám sát: Phương pháp huấn luyện hai pha giúp mô hình khởi tạo tốt hơn, giảm thiểu ảnh hưởng của dữ liệu nhiễu và tăng khả năng tổng quát hóa. Mô hình sau huấn luyện tự giám sát có thể sử dụng trọng số khởi tạo cho các thành phần chính như Image Feature Extractor, Tag Embedder, Multi-Head Attention và Transformer Encoder.

  5. Mở rộng cho dữ liệu video: Trong bài toán ETS với dữ liệu video, các mô hình trích xuất đặc trưng như R(2+1)D và SlowFast cho kết quả tốt hơn so với R3D và CSN. Việc áp dụng Temporal Shift Module (TSM) giúp giảm chi phí tính toán và tăng độ chính xác nhận dạng hành động trong video. Chiến lược lựa chọn khung hình phù hợp (ví dụ lấy mẫu ngẫu nhiên trong huấn luyện và Center Crop trong kiểm định) giúp cân bằng giữa hiệu quả và tốc độ xử lý.

Thảo luận kết quả

Nguyên nhân chính của sự cải thiện hiệu quả mô hình là nhờ vào việc kết hợp thông tin đa luồng (thẻ và ảnh) qua cơ chế Gating, giúp tận dụng tối đa đặc trưng từ cả hai nguồn dữ liệu. Hàm mất mát BCE-Dice khắc phục được vấn đề mất cân bằng dữ liệu phổ biến trong bài toán phân lớp đa nhãn. Việc sử dụng hàm mất mát phụ trợ giúp mô hình tránh bị lệ thuộc quá mức vào một luồng dữ liệu, từ đó cải thiện khả năng học của luồng còn lại.

So sánh với các nghiên cứu trước đây về tóm tắt văn bản và phân loại ảnh, nghiên cứu này mở rộng ứng dụng sang bài toán tóm tắt thẻ ảnh và video, đồng thời áp dụng các kỹ thuật học tự giám sát hiện đại để tận dụng dữ liệu chưa gán nhãn. Việc xử lý phần tử ngoại lai thông qua tăng cường dữ liệu là một điểm mới, giúp mô hình có khả năng chống nhiễu tốt hơn.

Dữ liệu có thể được trình bày qua biểu đồ so sánh giá trị F1 giữa các mô hình, biểu đồ hội tụ hàm mất mát qua các epoch, và bảng so sánh tỷ lệ ảnh bị ảnh hưởng bởi phần tử ngoại lai. Các biểu đồ này minh họa rõ ràng sự vượt trội của mô hình đề xuất và hiệu quả của các kỹ thuật bổ trợ.

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

  1. Triển khai mô hình MAGNeto trong hệ thống tìm kiếm ảnh và video: Tập trung cải thiện chỉ số F1 lên trên 0.92 trong vòng 6 tháng, do các nhóm phát triển AI và kỹ thuật thực hiện.

  2. Áp dụng chiến lược học tự giám sát cho các bài toán phân loại đa nhãn khác: Khuyến nghị các nhà nghiên cứu và doanh nghiệp sử dụng dữ liệu chưa gán nhãn để khởi tạo mô hình, giảm chi phí gán nhãn, hoàn thành trong 12 tháng.

  3. Tích hợp kỹ thuật Tag-adding và Tag-dropping trong quy trình tiền xử lý dữ liệu: Giảm thiểu ảnh hưởng phần tử ngoại lai, nâng cao độ chính xác mô hình, do bộ phận xử lý dữ liệu thực hiện, áp dụng ngay trong các dự án hiện tại.

  4. Mở rộng nghiên cứu và ứng dụng các mô hình trích xuất đặc trưng video như SlowFast và TSM: Tăng hiệu quả nhận dạng hành động trong video, giảm chi phí tính toán, dành cho các nhóm phát triển sản phẩm video thông minh, hoàn thành thử nghiệm trong 9 tháng.

  5. Phát triển hệ thống lựa chọn khung hình tự động cho dữ liệu video: Tối ưu hóa tốc độ xử lý và độ chính xác dự đoán, do nhóm nghiên cứu AI và kỹ thuật phần mềm phối hợp thực hiện, triển khai trong vòng 6 tháng.

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

  1. Nhà nghiên cứu và sinh viên ngành Toán Tin, Trí tuệ nhân tạo và Thị giác máy tính: Nắm bắt kiến thức về mô hình học sâu, Transformer, và các kỹ thuật học tự giám sát trong bài toán phân loại đa nhãn.

  2. Chuyên gia phát triển sản phẩm AI trong lĩnh vực xử lý ảnh và video: Áp dụng mô hình MAGNeto và các kỹ thuật tăng cường dữ liệu để cải thiện hiệu quả hệ thống tìm kiếm và phân loại.

  3. Doanh nghiệp cung cấp dịch vụ lưu trữ và quản lý dữ liệu đa phương tiện: Tối ưu hóa quy trình gán thẻ và tóm tắt thẻ cho ảnh và video, nâng cao trải nghiệm người dùng.

  4. Nhà phát triển phần mềm và kỹ sư dữ liệu: Học hỏi về chiến lược huấn luyện hai pha, xử lý mất cân bằng dữ liệu và phần tử ngoại lai, áp dụng trong các dự án thực tế.

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

  1. Bài toán Extractive Tags Summarization (ETS) là gì?
    ETS là bài toán thu gọn danh sách các thẻ gán cho ảnh hoặc video mà không tạo ra thẻ mới, nhằm giữ lại khả năng đại diện chính xác cho nội dung. Ví dụ, từ 50 thẻ gốc, mô hình chọn ra 10 thẻ quan trọng nhất.

  2. Tại sao sử dụng mô hình học sâu thay vì các phương pháp truyền thống?
    Mô hình học sâu như Transformer có khả năng học được mối quan hệ phức tạp giữa các thẻ và đặc trưng ảnh, xử lý hiệu quả dữ liệu lớn và đa dạng, vượt trội hơn các thuật toán cổ điển về độ chính xác và tốc độ.

  3. Làm thế nào để giải quyết vấn đề mất cân bằng dữ liệu trong bài toán ETS?
    Kết hợp hàm mất mát Binary Cross-Entropy với hàm Dice giúp giảm thiểu ảnh hưởng của lớp chiếm đa số, tăng độ nhạy với các thẻ ít xuất hiện, từ đó cải thiện độ chính xác tổng thể.

  4. Chiến lược học tự giám sát được áp dụng như thế nào?
    Mô hình được huấn luyện trước trên bài toán phát hiện phần tử ngoại lai bằng dữ liệu chưa gán nhãn, sau đó tinh chỉnh trên dữ liệu có gán nhãn, giúp mô hình khởi tạo tốt và tăng khả năng tổng quát hóa.

  5. Các mô hình trích xuất đặc trưng video nào được đánh giá hiệu quả nhất?
    SlowFast và Temporal Shift Module (TSM) được chứng minh có hiệu quả cao trong việc trích xuất đặc trưng video, cân bằng giữa độ chính xác và chi phí tính toán, phù hợp cho các ứng dụng nhận dạng hành động.

Kết luận

  • Đã phát triển thành công mô hình học sâu MAGNeto với kiến trúc đa luồng, kết hợp thông tin thẻ và ảnh, đạt giá trị F1 khoảng 0.92 trên bộ dữ liệu NUS-WIDE.
  • Áp dụng hàm mất mát BCE-Dice và hàm mất mát phụ trợ giúp cải thiện tốc độ hội tụ và độ chính xác mô hình.
  • Triển khai chiến lược học tự giám sát hai pha và kỹ thuật tăng cường dữ liệu Tag-adding/Tag-dropping để giảm thiểu ảnh hưởng phần tử ngoại lai.
  • Mở rộng bài toán ETS sang dữ liệu video với các mô hình trích xuất đặc trưng tiên tiến như SlowFast và TSM, đồng thời đề xuất chiến lược lựa chọn khung hình hiệu quả.
  • Khuyến nghị triển khai mô hình và kỹ thuật nghiên cứu trong các hệ thống tìm kiếm và quản lý dữ liệu đa phương tiện, đồng thời tiếp tục nghiên cứu mở rộng ứng dụng trong các lĩnh vực liên quan.

Tiếp theo, cần tiến hành thử nghiệm mở rộng trên các bộ dữ liệu đa dạng hơn và tối ưu hóa mô hình cho các ứng dụng thực tế. Mời các nhà nghiên cứu và chuyên gia trong lĩnh vực tham khảo và áp dụng các kết quả nghiên cứu này để phát triển các giải pháp AI tiên tiến hơn.