Tổng quan nghiên cứu
Ung thư đại trực tràng (Colorectal Cancer - CRC) hiện là một trong những nguyên nhân gây tử vong hàng đầu trên thế giới, xếp thứ 3 về tỷ lệ mắc mới và thứ 2 về tỷ lệ tử vong trong số các bệnh ung thư. Theo thống kê y tế quốc tế vào năm 2023, ước tính có khoảng 150.000 ca mắc mới và hơn 52.000 ca tử vong tại Hoa Kỳ. Tại Việt Nam, ung thư đại trực tràng liên tục nằm trong nhóm 10 bệnh ung thư phổ biến nhất, xếp thứ 4 ở nam giới và thứ 3 ở nữ giới, với tỷ lệ mắc chuẩn theo tuổi đạt khoảng 10,1 ca trên 100.000 dân. Việc phát hiện sớm các tổn thương tiền ung thư như polyp đại tràng đóng vai trò sống còn, nâng tỷ lệ sống sót sau 5 năm của bệnh nhân lên tới 90%. Dù vậy, trong thực tế chỉ có khoảng 40% ca bệnh được chẩn đoán ở giai đoạn sớm, và các bác sĩ nội soi có thể bỏ sót khoảng 25% số lượng polyp trong các quy trình kiểm tra thông thường.
Nhằm giải quyết bài toán cấp thiết này, các hệ thống trí tuệ nhân tạo (AI) và học sâu (Deep Learning) hỗ trợ chẩn đoán nội soi đã được phát triển, giúp giảm tỷ lệ bỏ sót polyp lên tới 50%. Tuy nhiên, việc huấn luyện mô hình phát hiện polyp đòi hỏi khối lượng lớn dữ liệu gán nhãn chính xác từ các chuyên gia y tế – một quy trình vô cùng tốn kém và khó mở rộng. Trong khi đó, các nguồn video nội soi chưa gán nhãn lại vô cùng dồi dào. Luận văn Thạc sĩ Khoa học Dữ liệu thực hiện tại Trường Công nghệ Thông tin và Truyền thông – Đại học Bách khoa Hà Nội vào tháng 10 năm 2023 đã tập trung nghiên cứu giải pháp phát hiện polyp đại tràng bằng phương pháp học bán giám sát (Semi-Supervised Object Detection - SSOD). Nghiên cứu đề xuất mô hình CenterNet++ kết hợp kỹ thuật Dense Target Producer (DTP) và cơ chế thích ứng ngưỡng động Threshold Epoch Adaptor (TEA), mang lại bước đột phá với tốc độ xử lý vượt mốc 200 khung hình/giây (FPS) và cải thiện vượt bậc độ chính xác khi chỉ sử dụng một lượng nhỏ dữ liệu gán nhã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 xây dựng trên nền tảng của bài toán thị giác máy tính và học bán giám sát hiện đại, tích hợp các lý thuyết cốt lõi sau:
- Lý thuyết phát hiện vật thể dựa trên bản đồ nhiệt (Heatmap-based Object Detection): Khác với các phương pháp truyền thống dựa trên hộp neo (Anchor-based) hoặc thuật toán triệt tiêu phi cực đại (Non-Maximum Suppression - NMS) vốn làm giảm tốc độ suy luận, mô hình dạng heatmap như CenterNet xem mỗi polyp là một điểm trung tâm trên bản đồ đặc trưng. Điểm tâm này được tạo ra thông qua hàm nhân Gaussian với độ lệch chuẩn thích ứng theo kích thước vật thể, cho phép mô hình huấn luyện và suy luận trực tiếp từ đầu đến cuối (End-to-End).
- Lý thuyết học bán giám sát và chưng cất tri thức (Knowledge Distillation SSOD): Áp dụng kiến trúc mạng kép Giáo viên - Học sinh (Teacher - Student). Trọng số của mô hình giáo viên được cập nhật liên tục từ mô hình học sinh theo phương pháp trung bình động hàm mũ (Exponential Moving Average - EMA) với hệ số suy giảm xác định, tạo ra các nhãn giả nhất quán cho dữ liệu chưa gán nhãn.
- Cơ chế tập trung đa tỉ lệ (Adaptive Scale Fusion - ASF) và khối tích chập nâng cao: Kết hợp các khối C2f, SPPF và hàm kích hoạt SiLU từ kiến trúc YOLOv8 nhằm tối ưu hóa khả năng trích xuất đặc trưng đa quy mô, giúp mô hình nhận diện chính xác các polyp có kích thước siêu nhỏ hoặc hình dạng phức tạp.
- Hàm mất mát kết hợp: Sử dụng Focal Loss tùy biến cho nhánh heatmap cùng với hàm mất mát chuẩn L1 cho các nhánh dự đoán độ lệch tâm (offset) và kích thước polyp (size).
Phương pháp nghiên cứu
- Nguồn dữ liệu và Cỡ mẫu: Nghiên cứu sử dụng bộ dữ liệu chuẩn hóa PolypsSet được tổng hợp từ 4 nguồn nội soi quốc tế uy tín, bao gồm MICCAI 2017 (38 video), CVC colon DB (15 video với 300 khung hình), GLRC (76 chuỗi video) và tập dữ liệu lớn KUMC từ Trung tâm Y tế Đại học Kansas gồm 80 video nội soi. Tổng số khung hình của tập KUMC lên tới 37.899 ảnh, được chia thành 28.773 ảnh huấn luyện, 4.254 ảnh kiểm định và 4.872 ảnh thử nghiệm độc lập, phân loại thành 2 nhóm bệnh học: polyp tuyến (adenomatous) và polyp tăng sản (hyperplastic).
- Phương pháp chọn mẫu bán giám sát: Để đánh giá năng lực học bán giám sát, nghiên cứu thực hiện kỹ thuật lấy mẫu ngẫu nhiên có phân tầng với các tỷ lệ dữ liệu có nhãn lần lượt là 1%, 5% và 10% trên toàn bộ tập huấn luyện; phần còn lại từ 90% đến 99% dữ liệu được giữ nguyên dưới dạng không gán nhãn.
- Quy trình phân tích và Huấn luyện: Mô hình được thiết lập huấn luyện trên 01 GPU NVIDIA RTX 3090 (24GB VRAM) với thuật toán tối ưu hóa AdamW, tốc độ học khởi tạo ở mức $10^{-3}$ và giảm dần về $10^{-5}$ theo lịch trình Cosine Annealing. Quá trình huấn luyện kéo dài 100 epoch cho kịch bản toàn giám sát và 50 epoch cho kịch bản bán giám sát. Nghiên cứu áp dụng chiến lược tăng cường dữ liệu phân tầng: tăng cường yếu (Weak Augmentations gồm lật ảnh, xoay, biến đổi tỷ lệ) cho mô hình giáo viên; tăng cường mạnh (Strong Augmentations gồm ColorJitter, Gaussian Blur, Dropout) kết hợp kỹ thuật Mosaic ghép 4 ảnh cho mô hình học sinh.
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 trên tập kiểm thử độc lập đã chứng minh những ưu thế vượt trội của phương pháp đề xuất qua các phát hiện nổi bật:
- CenterNet++ tối ưu hóa toàn diện về độ chính xác và tốc độ: Trong bài toán phát hiện đơn lớp có giám sát toàn bộ, mô hình CenterNet++ đạt độ chính xác trung bình $AP_{50}$ lên tới 89,2%, vượt qua mô hình CenterNet gốc 4,9% (đạt 84,3%) và vượt qua YOLOv8n 4,0% (đạt 85,2%). Đáng chú ý, mô hình đạt tốc độ suy luận vượt mốc 200 FPS với kích thước tham số siêu nhẹ chỉ 3,2 triệu tham số, giảm tới 86,7% dung lượng so với mức 24 triệu tham số của kiến trúc nguyên bản.
- Đột phá vượt bậc trong điều kiện khan hiếm dữ liệu nhãn: Khi chỉ sử dụng 1% dữ liệu có nhãn trong bài toán đơn lớp, phương pháp Dense Target Producer (DTP) đạt $AP_{50}$ là 59,4%, tạo ra mức tăng trưởng vượt bậc +16,8% so với mô hình học có giám sát thông thường chỉ đạt 42,6%. Khi nâng tỷ lệ nhãn lên 5% và 10%, DTP tiếp tục ghi nhận độ chính xác lần lượt là 73,1% và 80,2%, cao hơn từ 12,8% đến 15,6% so với mô hình baseline.
- Nâng cao năng lực phân loại bệnh học đa lớp: Đối với kịch bản phân loại chi tiết 2 nhóm polyp tuyến và polyp tăng sản, phương pháp DTP giúp gia tăng độ chính xác trung bình thêm 18,2% ở mức 1% dữ liệu nhãn, 14,6% ở mức 5% nhãn và 11,8% ở mức 10% nhãn so với phương pháp huấn luyện giám sát truyền thống.
- Hiệu quả rõ rệt của cơ chế thích ứng ngưỡng động và ghép ảnh: Các phân tích triệt tiêu (Ablation Studies) khẳng định việc tích hợp mô-đun kết hợp đặc trưng ASF cùng kỹ thuật tăng cường Mosaic đã đóng góp mức tăng thêm 6,0% $AP_{50}$ (từ 83,2% lên 89,2%). Cấu trúc xương sống tối ưu hóa từ YOLOv8n cũng chứng minh hiệu năng dẫn đầu khi đạt 69,8% mean $AP_{50}$ cho 2 lớp, vượt xa mức 66,5% của xương sống ResNet50 vốn có kích thước tham số lớn hơn gấp 7 lần (23,5 triệu tham số).
Thảo luận kết quả
Kết quả nghiên cứu đã giải quyết triệt để rào cản lớn nhất của các kỹ thuật gán nhãn giả dạng hộp (Pseudo-boxes) truyền thống. Trong các mô hình heatmap end-to-end, việc gán nhãn giả dạng bounding box thường dẫn đến hiện tượng trôi lệch tâm nghiêm trọng. Khi tâm giả bị lệch dù chỉ vài pixel, hàm Focal Loss sẽ triệt tiêu xác suất của điểm tâm thực tế về mức 0, khiến mô hình học sinh bị tích lũy sai số. Kỹ thuật Dense Target Producer (DTP) đã khắc phục hoàn toàn điểm yếu này bằng cách sử dụng trực tiếp ma trận bản đồ nhiệt dày đặc từ mô hình giáo viên, biến bản đồ này thành ma trận trọng số hướng dẫn việc học kích thước và độ lệch tâm ở các vùng lân cận.
Thêm vào đó, cơ chế ngưỡng động Threshold Epoch Adaptor (TEA) với các tham số điều khiển $\lambda_1 = 0,2$, $\lambda_2 = 0,2$ và chọn lọc $k=5$ điểm cực đại đã tự động nâng ngưỡng lọc nhãn song hành cùng tiến trình huấn luyện của mô hình. Dữ liệu thực nghiệm được lượng hóa chi tiết qua các đường cong Precision-Recall và bảng so sánh đa chỉ số ($AP_{50}$, $AP_{50:95}$, FPS, dung lượng tham số). Trực quan hóa hình ảnh cho thấy phương pháp mới đã loại bỏ hầu hết các trường hợp bỏ sót polyp ẩn dưới nếp gấp niêm mạc và giảm thiểu tối đa tình trạng nhận diện nhầm giữa polyp tăng sản lành tính với polyp tuyến có nguy cơ ác tính cao.
Đề xuất và khuyến nghị
Dựa trên các kết quả thực nghiệm xuất sắc, luận văn đưa ra 4 khuyến nghị và lộ trình triển khai cụ thể:
- Tích hợp mô hình vào thiết bị nội soi lâm sàng: Khuyến nghị các kỹ sư y sinh và chuyên gia AI tiến hành đóng gói mô hình CenterNet++ lên các bộ xử lý biên gắn trực tiếp vào máy nội soi tiêu hóa tại các bệnh viện. Mục tiêu duy trì độ trễ suy luận dưới 5 mili-giây (tương đương trên 200 FPS) trong thời gian 6 đến 12 tháng tới nhằm cung cấp chỉ dẫn thị giác theo thời gian thực cho bác sĩ trong lúc thao tác.
- Khai thác toàn diện nguồn dữ liệu video y tế chưa gán nhãn: Các cơ sở y tế và nhóm nghiên cứu nên thiết lập quy trình tự động thu thập luồng video nội soi hàng ngày để mở rộng khung học bán giám sát DTP. Đích đến là tận dụng 100% dữ liệu không nhãn tích lũy trong giai đoạn 2024–2025, giúp nâng cao độ ổn định của hệ thống trước nhiều dòng thiết bị nội soi khác nhau.
- Mở rộng cơ chế thích ứng ngưỡng động TEA sang các lĩnh vực chẩn đoán hình ảnh khác: Đề xuất các nhà khoa học dữ liệu ứng dụng thuật toán ngưỡng động TEA cho các bài toán phát hiện khối u phổi trên ảnh CT-scan hoặc tổn thương võng mạc tiểu đường, hướng tới mục tiêu cắt giảm tỷ lệ cảnh báo dương tính giả xuống dưới mức 5% trong vòng 12 tháng.
- Xây dựng hệ thống gán nhãn bán tự động hỗ trợ chuyên gia: Các viện nghiên cứu y khoa cần phát triển nền tảng học chủ động (Active Learning) kết hợp với nhãn giả tạo bởi DTP. Giải pháp này giúp định hướng chuyên gia y tế chỉ cần tập trung thẩm định các mẫu tổn thương phức tạp, giảm thiểu khoảng 70% thời gian và chi phí gán nhãn thủ công trong vòng 2 năm.
Đối tượng nên tham khảo luận văn
Nội dung và đóng góp của luận văn mang lại giá trị học thuật và ứng dụng thực tiễn cao cho 4 nhóm đối tượng chính:
- Nhà nghiên cứu và Học viên chuyên ngành Trí tuệ nhân tạo, Khoa học dữ liệu: Tài liệu là nguồn tham khảo chuyên sâu về phương pháp học bán giám sát cho mạng phát hiện vật thể dạng bản đồ nhiệt không dùng hộp neo (Anchor-free) và không dùng NMS, cung cấp các công thức toán học và thiết kế hàm mất mát chi tiết.
- Kỹ sư phát triển hệ thống AI Y tế (Healthcare AI Engineers): Cung cấp giải pháp tối ưu hóa kiến trúc mạng nơ-ron tích chập siêu nhẹ với 3,2 triệu tham số, giúp dễ dàng chuyển giao công nghệ và triển khai trên các hệ thống phần cứng y tế có tài nguyên tính toán giới hạn.
- Bác sĩ nội soi tiêu hóa và Chuyên gia y tế: Giúp đội ngũ y khoa hiểu rõ cơ chế vận hành của các mô hình hỗ trợ chẩn đoán tự động (CADe/CADx), từ đó nâng cao niềm tin và sự phối hợp hiệu quả với hệ thống AI nhằm giảm thiểu tỷ lệ bỏ sót 25% polyp trong thực hành lâm sàng.
- Các doanh nghiệp sản xuất phần mềm và thiết bị y tế: Tài liệu định hướng phương pháp tiếp cận kinh tế để phát triển các tính năng chẩn đoán thông minh thế hệ mới cho máy nội soi với chi phí đầu tư gán nhãn dữ liệu thấp nhất.
Câu hỏi thường gặp
Tại sao kiến trúc CenterNet++ lại đạt tốc độ xử lý vượt trội trên 200 FPS?
Mô hình CenterNet++ loại bỏ hoàn toàn các thành phần tính toán phức tạp như hộp neo (Anchor boxes) và thuật toán triệt tiêu phi cực đại (NMS). Bằng việc tinh gọn xương sống dựa trên các khối C2f và tích hợp mô-đun kết hợp đặc trưng ASF, mô hình giảm kích thước xuống chỉ còn 3,2 triệu tham số, cho phép suy luận siêu nhanh trên phần cứng tiêu chuẩn.
Phương pháp Dense Target Producer (DTP) giải quyết hạn chế gì của kỹ thuật gán nhãn giả truyền thống?
Kỹ thuật gán nhãn giả dạng hộp thông thường dễ gây ra sai lệch tâm vật thể, làm triệt tiêu giá trị cực đại của heatmap chuẩn. DTP khắc phục triệt để bằng cách truyền trực tiếp bản đồ nhiệt dày đặc từ mô hình giáo viên sang mô hình học sinh làm ma trận trọng số, giúp bảo toàn thông tin biên và kích thước polyp một cách liên tục và khả vi.
Cơ chế Threshold Epoch Adaptor (TEA) hoạt động như thế nào trong quá trình huấn luyện?
Thay vì sử dụng một ngưỡng lọc cố định dễ gây mất mẫu tốt hoặc nhận mẫu rác, TEA tự động nâng ngưỡng lọc nhãn giả theo sự gia tăng của số lượng epoch huấn luyện. Cơ chế này đảm bảo mô hình chỉ học các điểm tâm có độ tự tin cao nhất trong giai đoạn đầu và mở rộng dần không gian đặc trưng ở các epoch sau.
Khả năng hoạt động của mô hình ra sao khi lượng dữ liệu có nhãn chỉ chiếm 1%?
Trong điều kiện cực kỳ khan hiếm dữ liệu với chỉ 1% nhãn được giám sát, phương pháp DTP vẫn đạt mức $AP_{50}$ ấn tượng là 59,4%, tạo ra mức cải thiện vượt bậc +16,8% về độ chính xác so với mô hình huấn luyện có giám sát thông thường.
Tập dữ liệu PolypsSet được cấu thành từ những nguồn nào?
PolypsSet tổng hợp 4 bộ dữ liệu nội soi tiêu chuẩn gồm MICCAI 2017 (38 video), CVC colon DB (15 video), GLRC (76 video) và KUMC (80 video). Riêng tập dữ liệu KUMC đóng góp 37.899 khung hình được phân định rõ ràng thành 2 nhóm bệnh học: polyp tuyến tiền ung thư và polyp tăng sản lành tính.
Kết luận
- Đề xuất thành công kiến trúc CenterNet++ đạt độ chính xác $AP_{50}$ 89,2% với kích thước siêu nhẹ 3,2 triệu tham số và tốc độ xử lý vượt 200 FPS.
- Tiên phong ứng dụng phương pháp học bán giám sát (SSOD) vào mô hình phát hiện vật thể dạng bản đồ nhiệt end-to-end thông qua cơ chế Dense Target Producer.
- Xây dựng bộ điều chỉnh ngưỡng động Threshold Epoch Adaptor giúp kiểm soát chất lượng nhãn giả tối ưu theo từng giai đoạn huấn luyện.
- Đạt mức tăng trưởng độ chính xác vượt bậc từ 11,8% đến 18,2% trên tập dữ liệu chuẩn PolypsSet trong các kịch bản khan hiếm dữ liệu nhãn từ 1% đến 10%.
- Mở ra hướng đi mang tính ứng dụng thực tiễn cao cho việc nhúng các mô hình AI hỗ trợ chẩn đoán thời gian thực vào quy trình nội soi tiêu hóa tại các bệnh viện.
Luận văn đã đóng góp một khung giải pháp toàn diện cả về mặt lý thuyết mô hình lẫn thực nghiệm ứng dụng cho bài toán phát hiện tổn thương y tế bán giám sát. Trong các giai đoạn tiếp theo, nhóm nghiên cứu dự kiến mở rộng mô hình trên các chuỗi dữ liệu video spatio-temporal và thử nghiệm lâm sàng diện rộng. Các nhà nghiên cứu và đơn vị y tế quan tâm có thể tiếp cận mã nguồn và trọng số mô hình để phục vụ công tác nghiên cứu và triển khai thực tế.