Tổng quan nghiên cứu

Trong kỷ nguyên công nghiệp sản xuất thông minh, bài toán kiểm soát chất lượng sản phẩm bằng thị giác máy tính đang đối mặt với rào cản lớn: sự mất cân bằng dữ liệu nghiêm trọng khi số lượng mẫu sản phẩm lỗi trong thực tế chiếm tỷ lệ rất nhỏ, thường dưới 1% tổng sản lượng. Để giải quyết thách thức này, nghiên cứu "Phân lớp dữ liệu hình ảnh một lớp ứng dụng trong các hệ thống giám sát lỗi" của tác giả Trần Hậu Tuấn, dưới sự hướng dẫn của TS. Lê Hồng Trang tại Khoa Khoa học và Kỹ thuật Máy tính, Trường Đại học Bách Khoa – Đại học Quốc gia Thành phố Hồ Chí Minh (thực hiện từ ngày 01/03/2021 đến ngày 28/06/2021), đã tập trung nghiên cứu giải pháp phân lớp một lớp (One-Class Classification - OCC) tiên tiến nhằm phát hiện bất thường bề mặt sản phẩm công nghiệp.

Mục tiêu trọng tâm của đề tài là xây dựng mô hình trích xuất đặc trưng sâu kết hợp phân phối vùng để phát hiện lỗi chính xác mà chỉ cần huấn luyện trên tập dữ liệu hoàn toàn là sản phẩm chuẩn (không chứa lỗi). Phạm vi thực nghiệm của nghiên cứu được triển khai toàn diện trên bộ chuẩn dữ liệu quốc tế MVTec AD gồm 5.354 ảnh chất lượng cao thuộc 15 danh mục sản phẩm khác nhau. Kết quả nghiên cứu mang lại giá trị thực tiễn vượt bậc khi mô hình đề xuất đạt chỉ số diện tích dưới đường cong ROC (AUROC) trung bình lên đến 91,2% ở cấp độ toàn ảnh và đạt đỉnh 99,6% trên một số nhóm vật liệu chuyên biệt. Đồng thời, giải pháp chứng minh khả năng tối ưu hóa chi phí tính toán, cho phép phát hiện và khoanh vùng vị trí lỗi chính xác tới từng điểm ảnh, mở ra hướng đi bền vững cho các dây chuyền sản xuất tự động hóa hiện đạ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 được xây dựng dựa trên nền tảng lý thuyết học sâu (Deep Learning) và học máy thống kê, tập trung giải quyết bài toán phân lớp một lớp trong phát hiện dị thường thị giác. Khung lý thuyết chủ đạo bao gồm:

  1. Lý thuyết mạng nơ-ron tích chập (Convolutional Neural Network - CNN): Ứng dụng các kiến trúc trích xuất đặc trưng đa tầng như ResNet18 với các khối phần dư (Residual Blocks) giúp triệt tiêu hiện tượng suy biến gradient, và kiến trúc VGG16 gồm các tầng tích chập sâu nhằm thu nhận đặc trưng từ mức độ cục bộ đến toàn cục.
  2. Mô hình mô phỏng phân phối vùng PaDiM (Patch Distribution Modeling): Cơ chế biểu diễn đặc trưng không gian bằng cách chia nhỏ ảnh thành lưới tọa độ và gắn kết các vector nhúng đa mức kích hoạt từ các tầng ẩn khác nhau trong mạng nơ-ron.
  3. Lý thuyết thống kê đa biến và khoảng cách Mahalanobis: Mô hình hóa trạng thái bình thường của từng vùng ảnh bằng phân phối chuẩn Gaussian đa biến với vector kỳ vọng mẫu và ma trận hiệp phương sai. Khoảng cách Mahalanobis được sử dụng làm hàm mục tiêu tính toán điểm dị thường, giúp định lượng mức độ sai lệch của mẫu kiểm tra so với phân phối dữ liệu chuẩn có xét đến sự tương quan giữa các chiều đặc trưng.

Các khái niệm cốt lõi được định hình chặt chẽ gồm: Vùng tiếp nhận (Receptive Field), Bản đồ kích hoạt (Activation Map), Vector nhúng liên kết đa tầng (Multi-layer Embedding Vector) và Điểm dị thường cấp độ điểm ảnh (Pixel-level Anomaly Score).

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

Nghiên cứu sử dụng tập dữ liệu quy chuẩn MVTec AD gồm 5.354 hình ảnh màu, phân chia thành 3.629 ảnh huấn luyện chuẩn (chỉ gồm sản phẩm không lỗi) và 1.725 ảnh kiểm thử (bao gồm cả sản phẩm bình thường và 73 dạng khuyết tật bề mặt khác nhau). Cỡ mẫu thực nghiệm được thiết kế bao phủ 15 loại đối tượng, chia thành 5 nhóm kết cấu bề mặt (thảm, da, gỗ, gạch lát, lưới) và 10 nhóm vật thể rời (chai lọ, cáp điện, viên nhộng, đai ốc, bu lông, bàn chải, linh kiện bán dẫn).

Quy trình tiền xử lý và trích xuất đặc trưng được thực hiện qua các bước:

  • Ảnh đầu vào được chuẩn hóa kích thước về 256x256 điểm ảnh, sau đó cắt trung tâm (center crop) với kích thước 224x224 điểm ảnh theo chuẩn phân phối ImageNet.
  • Sử dụng mô hình ResNet18 tiền huấn luyện để trích xuất vector đặc trưng từ 3 khối tích chập đầu tiên, sau đó nối các vector lại với nhau.
  • Phương pháp chọn mẫu đặc trưng ngẫu nhiên (Random Dimensionality Selection) với kích thước 100 chiều trên ResNet18 và 200 chiều trên VGG16 được lựa chọn thay thế cho thuật toán PCA nhằm giảm thiểu 40% đến 60% thời gian tính toán mà vẫn đảm bảo tính đại diện thống kê.
  • Xây dựng tập dữ liệu thu nhỏ SMVTecAD (chỉ giữ lại 15% tập huấn luyện gốc, tương đương khoảng 550 ảnh với khoảng 35 ảnh cho mỗi loại sản phẩm) để đánh giá độ bền vững của mô hình trong điều kiện khan hiếm dữ liệu.

Phương pháp phân tích định lượng sử dụng độ đo Receiver Operating Characteristics (ROC) và Area Under the Curve (AUROC) trên cả hai cấp độ: phân lớp toàn bộ bức ảnh và phân vùng lỗi chi tiết theo từng điểm ảnh.

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

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

Quá trình thực nghiệm đối sánh giữa các mô hình và cấu hình dữ liệu mang lại 4 phát hiện khoa học quan trọng:

  1. Hiệu năng vượt trội của kiến trúc ResNet18: Trên toàn bộ 15 danh mục sản phẩm của tập dữ liệu MVTec AD, mô hình ResNet18 đạt điểm AUROC trung bình mức toàn ảnh là 91,2% (0,912), cao hơn đáng kể so với mức 85,7% (0,857) của mô hình VGG16, tương ứng mức chênh lệch hiệu năng cải thiện 5,5 điểm phần trăm.
  2. Độ chính xác tối ưu trên cấu trúc kết cấu đồng nhất: Mô hình đạt hiệu quả phân lớp gần như tuyệt đối trên các danh mục có hoa văn lặp lại hoặc tính đối xứng trục cao, dẫn đầu là sản phẩm da (Leather) đạt AUROC 99,6%, gỗ (Wood) đạt 99,1%, chai lọ (Bottle) đạt 98,6% và gạch lát (Tile) đạt 97,8%.
  3. Hạn chế trên các vật thể bất đối xứng và sai lệch góc chụp: Đối với các sản phẩm cơ khí nhỏ, góc chụp không cố định như đinh ốc (Screw), điểm AUROC ghi nhận mức thấp nhất là 74,1%, phản ánh độ nhạy cảm của phân phối Gaussian khi vị trí hình học của đối tượng bị dịch chuyển.
  4. Tính bền vững trên tập dữ liệu siêu nhỏ SMVTecAD: Khi giảm quy mô dữ liệu huấn luyện xuống chỉ còn 15% (khoảng 35 ảnh mỗi loại), mô hình vẫn duy trì AUROC trên 90,0% ở 10/15 danh mục sản phẩm, tiêu biểu như lớp thảm (Carpet), da (Leather), gạch (Tile) và chai lọ (Bottle), chứng minh khả năng học tổng quát xuất sắc mà không bị hiện tượng quá khớp (overfitting).

Thảo luận kết quả

Hiệu năng nổi bật của mô hình ResNet18 bắt nguồn từ việc liên kết các vector đặc trưng ở 3 mức độ khác nhau của mạng tích chập, kết hợp với việc ước lượng ma trận hiệp phương sai đầy đủ giúp lưu giữ trọn vẹn mối tương quan không gian giữa các vùng lân cận.

Dữ liệu kết quả thực nghiệm được trực quan hóa rõ ràng qua hai hình thức biểu diễn chính:

  • Bản đồ nhiệt dị thường (Anomaly Heatmaps): Các điểm ảnh có giá trị khoảng cách Mahalanobis cao được ánh xạ thành các dải màu ấm (đỏ, vàng), hiển thị sự trùng khớp chính xác trên 92% diện tích so với mặt nạ lỗi chuẩn (Ground Truth Mask).
  • Đồ thị đường cong ROC: Minh họa mối quan hệ đánh đổi giữa tỷ lệ dương tính thật (TPR) trên trục tung và tỷ lệ dương tính giả (FPR) trên trục hoành. Đường cong của các nhóm vật liệu như da và gỗ tiệm cận sát góc trên cùng bên trái của đồ thị, khẳng định mô hình duy trì tỷ lệ phát hiện lỗi cao trong khi tỷ lệ báo động giả cực thấp.

So sánh với các phương pháp tiếp cận dựa trên mô hình sinh như Autoencoder (AE) hay Generative Adversarial Networks (GAN), phương pháp phân phối vùng đề xuất loại bỏ hoàn toàn quá trình huấn luyện lại phức tạp (training-free), triệt tiêu rủi ro mô hình tự tái tạo luôn cả vùng khuyết tật, đồng thời giảm trên 70% thời gian hội tụ so với các mạng nơ-ron học sâu truyền thống.

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

Dựa trên kết quả nghiên cứu và các hạn chế thực tế đã được nhận diện, 4 nhóm giải pháp kỹ thuật mang tính hành động cao được đề xuất nhằm nâng cấp hệ thống:

  1. Nâng cấp thuật toán tiền xử lý và căn chỉnh không gian tự động: Đội ngũ kỹ sư R&D cần tích hợp các mạng nơ-ron định vị hoặc kỹ thuật biến đổi affine để chuẩn hóa góc quay và vị trí vật thể trước khi trích xuất đặc trưng. Mục tiêu nâng chỉ số AUROC trên nhóm sản phẩm cơ khí phức tạp như đinh ốc từ 74,1% lên trên 85,0% trong giai đoạn triển khai 6 tháng đầu năm.
  2. Tích hợp module tăng cường dữ liệu thích ứng (Data Augmentation): Áp dụng phương pháp xoay ngẫu nhiên từ âm 10 độ đến dương 10 độ kết hợp cắt ảnh ngẫu nhiên từ kích thước 256x256 về 224x224 điểm ảnh, giúp mô hình tăng cường khả năng chịu lỗi trước các biến động về độ rung lắc trên băng chuyền sản xuất thực tế.
  3. Triển khai ứng dụng giám sát công nghiệp chuyên dụng bằng C#: Nhóm phát triển phần mềm cần hiện thực hóa lõi thuật toán thành một ứng dụng desktop hoàn chỉnh trên nền tảng C# và thư viện đồ họa trực quan. Hệ thống cần đạt mục tiêu tốc độ xử lý dưới 50 mili-giây cho mỗi khung hình, tận dụng tối đa năng lực tăng tốc phần cứng của GPU công nghiệp.
  4. Mở rộng cơ chế học nửa giám sát (Semi-supervised Learning) và giải thích mô hình (Explainable AI): Các viện nghiên cứu và doanh nghiệp nên tiếp tục bổ sung các mẫu lỗi hiếm gặp đã gán nhãn để cập nhật phân phối thống kê, kết hợp cơ chế trực quan hóa kênh kích hoạt để hỗ trợ kỹ thuật viên kiểm định hiểu rõ nguyên nhân gây lỗi của từng linh kiện.

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

Nội dung và mã nguồn thực nghiệm của luận văn mang lại giá trị thực tiễn sâu sắc cho 4 nhóm đối tượng chính:

  1. Kỹ sư Trí tuệ nhân tạo và Thị giác máy tính (Computer Vision Engineers): Tiếp cận giải pháp phát hiện bất thường dựa trên mô hình nhúng không cần tinh chỉnh (training-free), giúp rút ngắn chu kỳ nghiên cứu và phát triển (R&D) các giải pháp thị giác trong nhà máy thông minh.
  2. Học viên cao học và Nhà nghiên cứu chuyên ngành Khoa học Máy tính: Khảo cứu tài liệu tham khảo chất lượng cao về phương pháp giải quyết bài toán mất cân bằng dữ liệu (Imbalanced Data) và phân lớp một lớp (OCC), làm tiền đề phát triển các thuật toán học sâu nâng cao.
  3. Kỹ sư Quản lý chất lượng (QA/QC) và Giám đốc vận hành nhà máy: Nắm bắt bức tranh tổng thể về khả năng tự động hóa khâu kiểm thử ngoại quan với độ chính xác đạt trên 91%, hỗ trợ ra quyết định đầu tư công nghệ nhằm cắt giảm chi phí nhân công kiểm tra thủ công.
  4. Nhà phát triển phần mềm công nghiệp: Tham khảo kiến trúc tích hợp giữa mô hình học sâu suy luận nhanh và ứng dụng điều khiển băng chuyền thực tế, giúp chuẩn hóa quy trình xuất bản giải pháp từ phòng thí nghiệm ra môi trường sản xuất.

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

Tại sao phân lớp một lớp lại là giải pháp tối ưu cho giám sát lỗi công nghiệp?
Trong dây chuyền sản xuất quy mô lớn, sản phẩm lỗi thường xuất hiện rất ngẫu nhiên và chiếm tỷ lệ cực thấp dưới 1% tổng số lượng. Phương pháp phân lớp một lớp cho phép hệ thống chỉ cần học các đặc trưng thống kê của sản phẩm chuẩn, giúp triển khai tự động hóa ngay lập tức mà không cần thu thập đủ hàng ngàn mẫu ảnh lỗi đa dạng.

Mô hình ResNet18 có ưu điểm gì vượt trội hơn VGG16 trong bài toán này?
ResNet18 sở hữu kiến trúc khối phần dư (Residual Blocks) giúp truyền dẫn thông tin đặc trưng đa mức hiệu quả mà không bị suy biến gradient. Trong thực nghiệm, ResNet18 đạt AUROC 91,2% với vector nhúng chỉ 100 chiều, vượt trội hơn VGG16 (85,7% AUROC) vốn đòi hỏi tới 200 chiều đặc trưng và tiêu tốn nhiều bộ nhớ tính toán hơn.

Hiệu năng mô hình thay đổi ra sao khi dữ liệu huấn luyện bị cắt giảm xuống 15%?
Khi thử nghiệm trên bộ dữ liệu nhỏ SMVTecAD với khoảng 35 ảnh huấn luyện cho mỗi loại, mô hình vẫn duy trì AUROC trung bình vượt mức 90,0% trên hầu hết các lớp sản phẩm có cấu trúc bề mặt đồng nhất như gỗ, da, thảm và chai lọ, khẳng định khả năng khái quát hóa vượt trội của phương pháp phân phối vùng.

Làm thế nào để hệ thống xác định chính xác vị trí lỗi trên bề mặt sản phẩm?
Mô hình tính toán khoảng cách Mahalanobis cho từng vector đặc trưng tại mỗi tọa độ lưới trên ảnh để tạo ra ma trận khoảng cách. Khi áp dụng ngưỡng quyết định, các điểm ảnh có giá trị khoảng cách vượt ngưỡng sẽ được ánh xạ thành bản đồ nhiệt (Heatmap), giúp khoanh vùng khuyết tật với độ chính xác tương đương mặt nạ lỗi chuẩn.

Đâu là thách thức lớn nhất khi đưa mô hình này vào môi trường sản xuất thực tế?
Thách thức lớn nhất nằm ở các sản phẩm có hình dạng bất đối xứng hoặc vị trí chụp bị xoay lệch tự do như đinh ốc (đạt AUROC 74,1%). Để khắc phục trong thực tế, hệ thống sản xuất cần bổ sung cơ chế cố định phôi cơ khí và tăng cường tiền xử lý căn chỉnh ảnh trước khi đưa vào mô hình suy luận.

Kết luận

  • Đề tài đã xây dựng thành công giải pháp phân lớp một lớp ứng dụng mạng tích chập ResNet18 và mô hình phân phối vùng PaDiM, giải quyết triệt để bài toán khan hiếm dữ liệu lỗi trong công nghiệp.
  • Kết quả thực nghiệm trên 5.354 ảnh thuộc tập dữ liệu MVTec AD đạt điểm AUROC trung bình 91,2% và đạt đỉnh 99,6% trên các sản phẩm có hoa văn kết cấu đồng nhất.
  • Đột phá kỹ thuật nằm ở việc ứng dụng khoảng cách Mahalanobis trên các vector đặc trưng ngẫu nhiên 100 chiều, giúp tối ưu hóa trên 50% thời gian xử lý và giải phóng dung lượng bộ nhớ.
  • Đề tài chứng minh tính khả thi cao trên tập dữ liệu nhỏ SMVTecAD (15% dữ liệu gốc) với độ chính xác duy trì trên 90,0%, sẵn sàng chuyển giao cho các ứng dụng thực tế.
  • Lộ trình phát triển tiếp theo tập trung vào việc hoàn thiện phần mềm giám sát thời gian thực bằng ngôn ngữ C# và mở rộng sang các tập dữ liệu nhiễu phức tạp.

Doanh nghiệp và các nhóm nghiên cứu quan tâm đến giải pháp tự động hóa kiểm soát chất lượng sản phẩm bằng thị giác máy tính có thể kết nối trực tiếp để tiếp nhận chuyển giao công nghệ, tối ưu hóa dây chuyền sản xuất và nâng cao năng lực cạnh tranh trong kỷ nguyên công nghiệp thông minh.