Tổng quan nghiên cứu

Trong quy trình sản xuất công nghiệp hiện đại, chi phí phát sinh từ việc kiểm định chất lượng sản phẩm thủ công và xử lý phế phẩm thường chiếm từ 15% đến 30% tổng ngân sách vận hành. Hơn 95% dữ liệu hình ảnh thu thập được trực tiếp trên dây chuyền là các sản phẩm đạt tiêu chuẩn (mẫu bình thường), trong khi các sản phẩm lỗi chỉ xuất hiện với tần suất cực kỳ hiếm hoi dưới 1%. Sự mất cân bằng dữ liệu nghiêm trọng này khiến các thuật toán phân loại nhiều lớp truyền thống hoàn toàn mất tác dụng do không thể thu thập đầy đủ mọi biến thể khuyết tật tiềm ẩn trong thực tế.

Nhằm giải quyết triệt để rào cản trên, luận văn thạc sĩ chuyên ngành Khoa học Máy tính mang tên "Phân loại dữ liệu một lớp và ứng dụng trong bài toán phát hiện bất thường" của học viên Nguyễn Văn Đức, dưới sự hướng dẫn khoa học của Phó Giáo sư Lê Hồng Trang tại Trường Đại học Bách Khoa, Đại học Quốc gia Thành phố Hồ Chí Minh, đã tập trung khai phá nhánh phân loại dữ liệu một lớp (One-Class Classification). Mục tiêu trọng tâm của đề tài là đề xuất kiến trúc chắt lọc tri thức ngược (Reverse Distillation) cải tiến dựa trên các đặc trưng học sâu, cho phép mô hình chỉ cần học từ dữ liệu sản phẩm hoàn hảo nhưng vẫn nhận diện và khoanh vùng chính xác mọi dạng bất thường.

Công trình được thực hiện trong giai đoạn từ tháng 9 năm 2021 đến tháng 12 năm 2022 và bảo vệ thành công vào tháng 2 năm 2023. Phạm vi nghiên cứu bao quát 15 danh mục sản phẩm công nghiệp trên tập dữ liệu chuẩn MVTec Anomaly Detection và 3 nhóm linh kiện trong tập dữ liệu BeanTech Anomaly Detection. Kết quả nghiên cứu mang lại ý nghĩa đột phá khi nâng cao độ chính xác nhận diện bất thường cấp mẫu đạt chỉ số AUROC trên 98,7% và độ chính xác định vị cấp điểm ảnh đạt trên 97,8%, hỗ trợ xử lý hình ảnh với thời gian thực chỉ 28 mili-giây mỗi khung hình.

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 3 trụ cột lý thuyết học sâu hiện đại: lý thuyết mạng nơ-ron tích chập (Convolutional Neural Networks - CNN), kiến trúc mạng phần dư ResNet-34 và cơ chế chắt lọc tri thức (Knowledge Distillation) dạng Teacher-Student ngược. Bên cạnh đó, tác giả phân tích và đối chiếu toàn diện với các mô hình kinh điển như mạng sinh đối kháng một lớp OCGAN và mạng tích chập OCCNN.

Khung lý thuyết vận dụng 4 khái niệm then chốt:

  • Phân loại dữ liệu một lớp: Cơ chế học máy mà không gian huấn luyện chỉ chứa duy nhất phân phối của lớp chuẩn, mô hình phải tự thiết lập ranh giới phân tách nhằm phát hiện mọi điểm dữ liệu ngoại lai.
  • Học không giám sát phát hiện bất thường: Phương pháp nhận diện các sai lệch về cấu trúc, bề mặt hoặc màu sắc mà không đòi hỏi bất kỳ hình ảnh lỗi nào trong quá trình huấn luyện.
  • Chắt lọc tri thức ngược: Cấu trúc truyền tải thông tin biểu diễn đặc trưng đa tầng từ mạng Teacher sang mạng Student thông qua một khối trung gian nén đặc trưng (Bottleneck).
  • Bản đồ nhiệt định vị bất thường: Ma trận sai số biểu diễn khoảng cách đặc trưng ở mức pixel giữa đầu ra của mạng Teacher và Student, cho phép trực quan hóa chính xác vị trí khuyết tật trên ảnh.

Về mô hình cốt lõi, mạng ResNet-34 với 34 tầng tích chập và tỷ lệ lỗi top-5 chỉ 3,6% trên ImageNet được chọn làm xương sống trích xuất đặc trưng tại các kênh 64, 128, 256 và 512, giải quyết triệt để hiện tượng triệt tiêu đạo hàm trong mạng sâu.

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

Nghiên cứu sử dụng nguồn dữ liệu thực nghiệm chuẩn mực từ 2 bộ benchmark quốc tế: MVTec Anomaly Detection gồm 5.354 ảnh độ phân giải cao từ 700x700 đến 1024x1024 pixels trên 15 danh mục sản phẩm (5 loại kết cấu bề mặt và 10 loại vật thể), cùng bộ dữ liệu BeanTech Anomaly Detection (BTAD) gồm 7.620 ảnh công nghiệp chia đều cho 3 dòng sản phẩm thực tế (2.540 ảnh mỗi nhóm).

Phương pháp chọn mẫu tuân thủ nguyên tắc phân tầng có chủ đích: 100% dữ liệu tập huấn luyện là ảnh sản phẩm hoàn hảo không có lỗi; tập kiểm thử độc lập bao gồm cả ảnh bình thường và ảnh chứa các dạng lỗi vật lý như trầy xước, nứt vỡ, biến màu, thủng lỗ kèm nhãn mặt nạ phân đoạn mức pixel.

Luận văn lựa chọn phương pháp phân tích chắt lọc tri thức ngược cải tiến kết hợp khối Bottleneck trên nền tảng PyTorch và TensorFlow. Lý do lựa chọn giải pháp này là vì cơ chế chắt lọc ngược ngăn chặn hiệu quả hiện tượng rò rỉ thông tin dư thừa từ mạng Teacher sang mạng Student, đồng thời triệt tiêu nguy cơ mạng Student tự ý khái quát hóa và sao chép luôn cả các đặc trưng bất thường chưa từng gặp. Toàn bộ chu trình thiết kế thuật toán, tối ưu siêu tham số với tốc độ học từ 0,001 đến 0,005 và thực nghiệm đánh giá được triển khai chặt chẽ trong suốt 15 tháng nghiên cứu.

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 đa tầng trên các bộ dữ liệu chuẩn quốc tế đã mang lại 4 phát hiện khoa học quan trọng:

Thứ nhất, mô hình chắt lọc ngược cải tiến đạt hiệu năng phát hiện bất thường cấp mẫu với chỉ số AUROC trung bình lên tới 98,7% trên tập dữ liệu MVTec AD. Kết quả này vượt trội hơn 4,2% so với mô hình sinh đối kháng OCGAN (đạt 94,5%) và cao hơn 6,8% so với phương pháp phân biệt OCCNN.

Thứ hai, ở cấp độ định vị khuyết tật từng điểm ảnh (pixel level), mô hình xác lập kỷ lục với AUROC đạt 97,8% và chỉ số bao phủ vùng lỗi (Per-Region Overlap - PRO) đạt 93,4% trên cả 15 danh mục sản phẩm, cải thiện 3,5% so với mô hình chắt lọc lũy tiến KDGAN.

Thứ ba, trên tập dữ liệu công nghiệp thực tế BTAD, thuật toán duy trì độ tin cậy xuất sắc với AUROC phân loại đạt 96,5% trên nhóm sản phẩm 1, 95,8% trên nhóm sản phẩm 2 và 97,2% trên nhóm sản phẩm 3, cùng chỉ số AUROC định vị điểm ảnh luôn vượt ngưỡng 96,0%.

Thứ tư, thuật toán tối ưu hóa thời gian tính toán vượt bậc khi đạt tốc độ suy luận 28 mili-giây mỗi bức ảnh kích thước 256x256 pixels, tương đương thông lượng xử lý 35 khung hình mỗi giây (FPS), giúp tiết kiệm 40% tài nguyên bộ nhớ GPU so với các kiến trúc tự mã hóa đa tầng phức tạp.

Thảo luận kết quả

Nguyên nhân cốt lõi giúp mô hình đề xuất đạt được độ chính xác vượt trội nằm ở kiến trúc khối Bottleneck nén không gian đặc trưng. Khi ảnh bình thường đi qua hệ thống, mạng Student dễ dàng khôi phục lại các đặc trưng tương đồng với mạng Teacher do đã được tối ưu hóa hàm mất mát cosine trong quá trình học. Ngược lại, khi xuất hiện vùng ảnh chứa khuyết tật, mạng Student do không có tri thức về dạng lỗi này nên không thể tái tạo lại biểu diễn của Teacher, tạo ra độ lệch khoảng cách đặc trưng rất lớn tại đúng tọa độ bất thường.

So sánh với các nghiên cứu tiền nhiệm sử dụng Autoencoder truyền thống thường gặp lỗi tái tạo hình ảnh bị nhòe mờ hoặc vô tình tái tạo luôn cả các khuyết tật nhỏ, mô hình chắt lọc ngược bảo toàn trọn vẹn chi tiết góc cạnh nhờ giữ nguyên đặc trưng ở các tầng conv1, conv2 và conv3 của ResNet-34.

Dữ liệu thực nghiệm được trực quan hóa rõ nét qua các bảng đối chuẩn đa chỉ số và biểu đồ đường cong ROC. Trên đồ thị ROC, đường biểu diễn hiệu năng của mô hình chắt lọc ngược luôn tiệm cận sát góc trên cùng bên trái với diện tích dưới đường cong lớn nhất. Bản đồ nhiệt (Heatmap) tạo ra từ sai số đặc trưng hiển thị độ tương đồng ranh giới khuyết tật trùng khớp tới 95% so với nhãn mặt nạ lỗi chuẩn của nhà sản xuất, chứng minh khả năng giải thích trực quan và độ tin cậy cao của mô hình.

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

Dựa trên các kết quả đạt được, luận văn đưa ra 4 khuyến nghị hành động mang tính ứng dụng thực tiễn cao:

Thứ nhất, chuẩn hóa hệ thống thu nhận hình ảnh đầu vào tại nhà máy. Doanh nghiệp sản xuất và các kỹ sư thị giác máy tính cần thiết lập hệ thống chiếu sáng công nghiệp đồng nhất cùng camera độ phân giải từ 1080p trở lên, nhằm giảm thiểu 25% độ nhiễu quang học trong vòng 3 tháng đầu triển khai dây chuyền.

Thứ hai, tối ưu hóa mô hình phục vụ triển khai thời gian thực bằng TensorRT. Đội ngũ phát triển phần mềm AI cần chuyển đổi mô hình ResNet-34 sang định dạng bán chính xác FP16 hoặc INT8, hướng tới mục tiêu rút ngắn thời gian xử lý xuống dưới 15 mili-giây mỗi ảnh và nâng thông lượng kiểm tra lên 60 FPS trong lộ trình 6 tháng.

Thứ ba, tích hợp thuật toán lên các thiết bị tính toán biên (Edge AI). Bộ phận tự động hóa tại các khu công nghiệp cần phối hợp triển khai mô hình trực tiếp trên các vi xử lý nhúng như NVIDIA Jetson để duy trì độ chính xác AUROC trên 95% trong điều kiện hoạt động liên tục 24/7, thực hiện trong vòng 9 tháng.

Thứ tư, thiết lập cơ chế học tăng cường trực tuyến (Online Continual Learning). Các nhóm nghiên cứu và chuyên gia vận hành hệ thống cần xây dựng quy trình tự động tái hiệu chỉnh trọng số khối Bottleneck theo chu kỳ 30 ngày một lần, đảm bảo mô hình nhanh chóng thích nghi khi mẫu mã sản phẩm thay đổi mà không làm suy giảm hiệu năng trên các dòng sản phẩm hiện hữu.

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

Công trình nghiên cứu mang lại giá trị học thuật và ứng dụng chuyên sâu cho 4 nhóm đối tượng trọng điểm:

  1. Học viên cao học và nghiên cứu sinh ngành Khoa học Máy tính: Luận văn cung cấp phương pháp luận nghiên cứu chuẩn mực về học một lớp, kỹ thuật xây dựng mô hình chắt lọc tri thức ngược và quy trình thiết lập thực nghiệm bài bản trên các tập dữ liệu quốc tế.
  2. Kỹ sư thị giác máy tính và kỹ sư MLOps: Tiếp cận giải pháp kiến trúc mạng tối ưu để triển khai trực tiếp vào các bài toán kiểm tra ngoại quan tự động (Automated Optical Inspection) cho bo mạch điện tử, sản phẩm may mặc, bao bì và linh kiện cơ khí chính xác với độ trễ dưới 30 mili-giây.
  3. Giám đốc công nghệ và Quản lý chất lượng (QA/QC) tại các nhà máy: Nắm bắt cơ sở khoa học và bài toán hiệu quả kinh tế để tự tin ứng dụng AI thay thế 70% khâu kiểm tra thủ công, giảm thiểu rủi ro lọt sản phẩm lỗi ra thị trường.
  4. Chuyên gia phân tích dữ liệu và an toàn thông tin: Khai thác tư duy phân loại một lớp để mở rộng sang các bài toán phát hiện gian lận thẻ tín dụng, phát hiện tấn công mạng zero-day hoặc chẩn đoán tổn thương y khoa trên ảnh chụp X-quang và MRI.

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

Phân loại dữ liệu một lớp khác biệt như thế nào so với phân loại nhị phân truyền thống? Phân loại nhị phân đòi hỏi tập huấn luyện phải có đầy đủ dữ liệu của cả 2 lớp bình thường và lỗi. Trong khi đó, phân loại một lớp chỉ cần huấn luyện trên 100% ảnh sản phẩm chuẩn. Trong thực tế công nghiệp, doanh nghiệp rất dễ dàng thu thập hàng nghìn mẫu hoàn hảo nhưng không thể lường trước mọi dạng hư hỏng tiềm ẩn.

Tại sao mô hình chắt lọc tri thức ngược lại vượt trội hơn các mô hình Autoencoder thông thường? Mô hình Autoencoder truyền thống thường tạo ra ảnh tái tạo bị nhòe và có xu hướng tái tạo luôn cả các lỗi nhỏ. Mô hình chắt lọc ngược khắc phục điều này bằng cách đóng băng mạng Teacher để trích xuất đặc trưng chuẩn xác, sau đó nén qua khối Bottleneck khiến mạng Student không thể tái tạo lại các vùng bất thường, làm sai số tại điểm lỗi tăng vọt trên 300%.

Các chỉ số AUROC và PRO có ý nghĩa cụ thể như thế nào trong bài toán phát hiện bất thường? Chỉ số AUROC đo lường diện tích dưới đường cong ROC giữa độ nhạy và tỷ lệ dương tính giả, phản ánh năng lực phân biệt chính xác mẫu hỏng ở cấp độ toàn bức ảnh đạt trên 98%. Chỉ số PRO đánh giá độ chính xác khi khoanh vùng khuyết tật ở cấp độ từng nhóm điểm ảnh, đảm bảo không bỏ sót các vết nứt siêu nhỏ từ 5 đến 10 pixels.

Mô hình trong luận văn có thể chạy trực tiếp trên dây chuyền sản xuất công nghiệp thực tế không? Mô hình hoàn toàn đáp ứng tốt môi trường công nghiệp nhờ tốc độ suy luận nhanh đạt 28 mili-giây mỗi ảnh trên các dòng GPU phổ thông. Với thông lượng 35 khung hình mỗi giây, hệ thống kiểm soát hoàn hảo các sản phẩm di chuyển trên băng chuyền tự động với vận tốc từ 0,5 đến 1 mét mỗi giây mà không gây tắc nghẽn.

Khi nhà máy thay đổi mẫu mã sản phẩm mới thì có cần huấn luyện lại toàn bộ mô hình không? Không cần huấn luyện lại từ đầu vì mạng trích xuất đặc trưng Teacher đã được tiền huấn luyện vững chắc trên ImageNet. Khi có sản phẩm mới, kỹ sư chỉ cần thu thập khoảng 200 đến 500 ảnh sản phẩm chuẩn và tinh chỉnh lại khối Bottleneck cùng mạng Student trong khoảng 1 đến 2 giờ trên máy trạm thông thường.

Kết luận

  • Đề tài giải quyết thành công bài toán mất cân bằng dữ liệu nghiêm trọng trong công nghiệp thông qua phương pháp phân loại dữ liệu một lớp.
  • Đề xuất kiến trúc chắt lọc tri thức ngược cải tiến kết hợp mạng ResNet-34 và khối Bottleneck, khắc phục hoàn toàn nhược điểm của các mạng Autoencoder và GAN truyền thống.
  • Xác lập hiệu năng ấn tượng với chỉ số AUROC phân loại đạt 98,7% trên tập MVTec AD và duy trì trên 96,0% trên bộ dữ liệu thực tế BTAD.
  • Khả năng định vị khuyết tật ở mức điểm ảnh đạt chỉ số PRO 93,4%, cung cấp bản đồ nhiệt trực quan giúp giải thích rõ ràng vị trí và hình dạng lỗi.
  • Đạt tốc độ xử lý thời gian thực 35 FPS với thời gian suy luận 28 mili-giây mỗi ảnh, sẵn sàng tích hợp vào dây chuyền sản xuất thông minh.

Đóng góp khoa học cốt lõi của tác giả Nguyễn Văn Đức đã được đúc kết và công bố trong bài báo khoa học quốc tế "An Improved Reverse Distillation Model for Unsupervised Anomaly Detection". Lộ trình nghiên cứu tiếp theo trong 12 tháng tới sẽ tập trung mở rộng giải thuật sang xử lý dữ liệu đám mây điểm 3D và tối ưu hóa nhúng chip phần cứng chuyên dụng. Luận văn là tài liệu tham khảo quý giá dành cho các nhà nghiên cứu và kỹ sư đang tìm kiếm giải pháp thị giác máy tính đột phá cho bài toán phát hiện bất thường.