Tổng quan nghiên cứu

Thị giác máy tính và học sâu đang tạo ra những bước chuyển mang tính cách mạng trong xử lý ảnh y tế và sinh trắc học. Tuy nhiên, các bài toán thực tế vẫn đối mặt với nhiều thách thức nghiêm trọng về độ chính xác và khả năng khái quát hóa. Trong nhận diện khuôn mặt đa góc nghiêng, sự thay đổi tư thế chụp làm suy giảm đáng kể hiệu năng của các hệ thống giám sát. Cùng với đó, trong lĩnh vực y tế, ung thư đại trực tràng là một trong những nguyên nhân gây tử vong hàng đầu thế giới; theo các thống kê lâm sàng, tỷ lệ bỏ sót polyp trong các ca nội soi đại tràng dao động từ 20% đến 47%, đặc biệt là đối với các tổn thương phẳng hoặc có kích thước dưới 10 mm. Cứ mỗi 1% tỷ lệ phát hiện u tuyến được cải thiện sẽ giúp giảm tương ứng 3% nguy cơ tiến triển thành ung thư đại trực tràng ác tính.

Xuất phát từ thực tiễn trên, luận văn thạc sĩ chuyên ngành Khoa học Máy tính tại Đại học Bách Khoa Hà Nội (hoàn thành vào tháng 04/2021) do học viên Trần Quang Chung thực hiện dưới sự hướng dẫn của Tiến sĩ Đinh Viết Sang đã tập trung nghiên cứu đề tài cải tiến kiến trúc mạng U-Net và mở rộng các ứng dụng thực tiễn. Mục tiêu trọng tâm của công trình là thiết kế các kiến trúc mạng nơ-ron tích chập chuyên sâu nhằm giải quyết hai bài toán then chốt: tái tạo khuôn mặt trực diện từ ảnh góc nghiêng để phục vụ nhận diện bất biến tư thế, và tự động phân vùng chính xác vùng polyp trên ảnh nội soi đường tiêu hóa.

Phạm vi thực nghiệm của nghiên cứu được triển khai toàn diện trên các tập dữ liệu tiêu chuẩn quốc tế quy mô lớn, bao gồm bộ dữ liệu Multi-PIE với 755.370 hình ảnh của 337 cá nhân dưới 15 góc chụp khác nhau và 19 điều kiện ánh sáng, cùng các tập dữ liệu nội soi đại tràng lâm sàng chuẩn như Kvasir-SEG và CVC-ClinicDB. Kết quả nghiên cứu đã tạo ra ba biến thể kiến trúc mạng mới (ResCUNet, Attention ResCUNet và Attention ResCUNeSt), mang lại sự cải thiện vượt bậc với chỉ số tương đồng mDice và mIoU tăng từ 2,1% đến 4,6% so với các mô hình tiêu chuẩn hiện hành, mở ra giải pháp ứng dụng thiết thực cho các hệ thống chẩn đoán y tế hỗ trợ máy tính và công nghệ an ninh thông minh.

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

Khung lý thuyết áp dụng

Luận văn xây dựng nền tảng dựa trên sự kết hợp hài hòa giữa các lý thuyết học sâu tiên tiến và kỹ thuật xử lý ảnh số chuyên sâu:

  • Kiến trúc mạng U-Net và Mạng ghép đôi (Coupled U-Nets): U-Net là kiến trúc mạng mã hóa - giải mã đối xứng kinh điển, trong đó các liên kết tắt (skip connections) truyền trực tiếp các đặc trưng cục bộ mức thấp từ bộ mã hóa sang bộ giải mã. Để tăng cường khả năng biểu diễn đặc trưng phức tạp, cấu trúc Coupled U-Nets kết nối hai mạng U-Net liên tiếp nhằm tinh chỉnh dần kết quả phân đoạn và thúc đẩy việc tái sử dụng đặc trưng qua các tầng mạng sâu.
  • Lý thuyết học phần dư và Khung kiến trúc ResNeSt (Split-Attention Networks): Nhằm khắc phục hiện tượng suy thoái gradient khi tăng độ sâu mạng, luận văn tích hợp cấu trúc phần dư từ họ mạng ResNet. Đặc biệt, kiến trúc ResNeSt được khai thác với cơ chế Split-Attention, phân chia các kênh đặc trưng thành nhóm cơ sở kết hợp siêu tham số Radix, cho phép mô hình học trọng số chú ý động giữa các nhánh không gian và kênh khác nhau.
  • Mạng sinh đối kháng (GAN) kết hợp Cơ chế Cổng chú ý (Attention Gates): Mô hình sinh đối kháng gồm bộ tạo (Generator) và hai bộ phân biệt (Global Discriminator và Local Discriminator) giúp tái tạo bản đồ UV map đầy đủ của khuôn mặt với độ chân thực cao. Cổng chú ý (Attention Gates) tự động lọc bỏ các phản hồi nền không liên quan và tập trung vào các vùng cấu trúc giải phẫu trọng yếu.
  • Hàm mất mát Tversky và Bảo toàn danh tính (Identity Preserving): Sử dụng hàm mất mát Tversky Loss có thể tinh chỉnh các hệ số phạt để tối ưu hóa sự cân bằng giữa độ chính xác (Precision) và độ nhạy (Recall). Kết hợp với đó là mô-đun trích xuất vector đặc trưng 512 chiều từ mạng tiền huấn luyện FaceNet trên 3,31 triệu ảnh để đảm bảo tính nhất quán danh tính sau tái tạo.

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

Nghiên cứu sử dụng quy trình thực nghiệm nghiêm ngặt với các thông số cụ thể:

  • Nguồn dữ liệu và Cỡ mẫu: Trong bài toán khuôn mặt, mô hình được huấn luyện trên cơ sở dữ liệu Multi-PIE (755.370 ảnh, 15 góc máy đặt cách nhau 15 độ) kết hợp các tập kiểm thử LFW, CPLFW và CFP. Trong bài toán y tế, các bộ dữ liệu nội soi chuẩn gồm Kvasir-SEG (1.000 ảnh) và CVC-ClinicDB (612 ảnh) được sử dụng để đánh giá năng lực phân vùng tổn thương.
  • Phương pháp chọn mẫu và Phân chia dữ liệu: Dữ liệu được chuẩn hóa về kích thước 512 × 512 pixel. Phương pháp kiểm thử chéo 5 lớp (5-fold cross-validation) được áp dụng đồng thời với việc thiết lập 6 kịch bản huấn luyện độc lập (Scenario 1 đến Scenario 6), tách biệt hoàn toàn giữa dữ liệu huấn luyện và dữ liệu kiểm thử từ các trung tâm y tế khác nhau để kiểm tra độ khái quát hóa xuyên tập dữ liệu.
  • Lý do lựa chọn phương pháp phân tích: Thuật toán tối ưu hóa Mini-batch Gradient Descent kết hợp với hàm mất mát kết hợp đa thành phần (gồm mất mát tái tạo, mất mát đối kháng, mất mát phụ trợ Auxiliary Loss và mất mát Tversky) được lựa chọn nhằm tối đa hóa tốc độ hội tụ và giải quyết triệt để hiện tượng mất cân bằng nghiêm trọng giữa diện tích polyp nhỏ với toàn bộ nền niêm mạc đường ruột.

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á định lượng và định tính trên các tập chuẩn benchmark đã ghi nhận những phát hiện đột phá:

  • Khả năng tái tạo bản đồ khuôn mặt 3D vượt trội: Biến thể Attention ResCUNet-GAN đã hoàn thiện thành công các vùng bản đồ UV bị che khuất do góc nghiêng khuôn mặt. So với mô hình cơ sở UV-GAN, hình ảnh trực diện được tổng hợp đạt độ chân thực tự nhiên, không xuất hiện hiện tượng biến dạng cục bộ ở vùng mắt hay khóe miệng, đồng thời bảo toàn nguyên vẹn danh tính chủ thể với khoảng cách không gian embedding 512 chiều đạt mức tối ưu.
  • Nâng cao độ chính xác nhận diện khuôn mặt đa tư thế: Khi tích hợp mô hình tái tạo khuôn mặt vào hệ thống nhận diện, độ chính xác xác thực trên các tập dữ liệu thách thức CPLFW và CFP ghi nhận mức tăng từ 1,5% đến 3,8% so với việc sử dụng trực tiếp các ảnh góc nghiêng ban đầu.
  • Đột phá về chỉ số phân đoạn tổn thương polyp y tế: Trên hai tập dữ liệu kiểm thử Kvasir-SEG và CVC-ClinicDB, mô hình Attention ResCUNeSt-101 thiết lập chuẩn hiệu năng mới với chỉ số mDice đạt trên 90% và mIoU vượt trội hơn từ 2,1% đến 4,6% khi đặt cạnh các mô hình nổi tiếng như UNet, ResUNet++, Double-UNet và PraNet.
  • Khả năng định vị chính xác tổn thương kích thước nhỏ: Cơ chế Split-Attention kết hợp với cổng chú ý Attention Gate cấp 9 (S9) đã triệt tiêu hiệu quả các vùng nhiễu phản quang của dịch nhầy và nếp gấp niêm mạc, giúp phát hiện chính xác các polyp có đường kính nhỏ dưới 10 mm mà các phương pháp truyền thống thường bỏ sót.

Thảo luận kết quả

Sự vượt trội của các mô hình đề xuất bắt nguồn từ việc thiết kế luồng truyền thông tin đa cấp thông minh. Việc kết hợp liên kết tắt giữa hai mạng U-Net liên tiếp giúp bộ giải mã của mạng thứ hai tiếp nhận đồng thời cả đặc trưng ngữ nghĩa mức cao và các chi tiết cấu trúc cục bộ mức thấp. Việc bổ sung nhánh tổn thất phụ trợ (Auxiliary Loss) tại đầu ra của mạng U-Net thứ nhất đóng vai trò then chốt trong việc duy trì dòng gradient ổn định, ngăn chặn hiện tượng triệt tiêu gradient ở các tầng nơ-ron sâu.

Trong phần trình bày kết quả, dữ liệu thực nghiệm được trực quan hóa chi tiết thông qua các hệ thống đồ thị và bảng biểu chuyên dụng:

  • Biểu đồ đường cong ROC và PR (Precision-Recall): Đồ thị trung bình qua 5 lớp kiểm thử chéo thể hiện diện tích dưới đường cong (AUC) của mô hình Attention ResCUNeSt-101 tiệm cận sát góc trên bên trái, duy trì độ nhạy vượt trội ở mọi ngưỡng phân loại so với PraNet và ResUNet++.
  • Bản đồ nhiệt kích hoạt chú ý (Attention Heatmaps): Các hình ảnh trực quan hóa phản ánh rõ nét sự tập trung tín hiệu; vùng màu đỏ rực bao phủ chính xác ranh giới của tổn thương polyp, trong khi các vùng niêm mạc bình thường bị ức chế hoàn toàn.
  • Bảng đối sánh đa tiêu chí: Thể hiện rõ ràng các chỉ số Recall, Precision, mDice và mIoU qua 6 kịch bản thử nghiệm độc lập, minh chứng tính ổn định của mô hình khi huấn luyện trên tập dữ liệu CVC-Colon và kiểm thử trực tiếp trên ETIS-Larib.

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

Dựa trên các kết quả đạt được, luận văn đưa ra 4 nhóm đề xuất ứng dụng và định hướng kỹ thuật mang tính khả thi cao:

  1. Tích hợp mô hình Attention ResCUNeSt vào hệ thống hỗ trợ nội soi theo thời gian thực: Đội ngũ kỹ sư phần mềm y tế phối hợp với các trung tâm nội soi cần đóng gói mô hình thành mô-đun CAD (Computer-Aided Diagnosis). Mục tiêu kỹ thuật cần đạt là duy trì tốc độ xử lý trên 30 khung hình/giây (fps) với độ trễ phản hồi dưới 35 mili-giây, triển khai thử nghiệm thực tế trong lộ trình từ 6 đến 12 tháng.
  2. Chuẩn hóa quy trình tiền xử lý phục hồi khuôn mặt 3D trong an ninh giám sát: Khuyến nghị các đơn vị phát triển giải pháp an ninh sinh trắc học tích hợp mạng 3DDFA và Attention ResCUNet-GAN vào luồng camera giám sát công cộng. Giải pháp này hướng tới việc nâng cao tỷ lệ nhận diện chính xác đối tượng nghiêng góc từ 45 đến 90 độ lên mức trên 98,5% trong vòng 6 tháng tới.
  3. Mở rộng việc áp dụng hàm mất mát Tversky Loss kết hợp Auxiliary Loss cho các bài toán phân đoạn ảnh y tế khác: Các viện nghiên cứu y sinh học nên ứng dụng cấu trúc hàm tổn thất này vào các dạng ảnh cộng hưởng từ (MRI) não hoặc CT phổi nhằm tăng chỉ số phân đoạn ranh giới tổn thương thêm ít nhất 3% trong giai đoạn 2024-2026.
  4. Thực hiện nén mô hình và lượng tử hóa trọng số (Quantization): Nhóm nghiên cứu hệ thống nhúng cần áp dụng kỹ thuật chưng cất tri thức (Knowledge Distillation) nhằm giảm từ 40% đến 50% kích thước bộ nhớ mô hình, tạo điều kiện thuận lợi để nạp thuật toán trực tiếp vào các vi xử lý phần cứng biên của máy nội soi cầm tay trong vòng 9 tháng.

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

Tài liệu học thuật này mang lại giá trị tham khảo sâu sắc cho 4 nhóm đối tượng trọng tâm:

  • Học viên cao học và Nghiên cứu sinh ngành Trí tuệ Nhân tạo / Khoa học Máy tính: Luận văn là tài liệu tham khảo mẫu mực về phương pháp cải tiến các kiến trúc mạng nơ-ron tích chập sâu, kỹ thuật ghép nối mạng Coupled Networks, cơ chế Split-Attention và phương pháp thiết kế hàm mất mát kết hợp cho các bài toán thị giác máy tính phức tạp.
  • Bác sĩ chuyên khoa tiêu hóa và Chuyên viên y tế nội soi: Tài liệu giúp các chuyên gia lâm sàng nắm bắt cơ chế hoạt động của các hệ thống AI hỗ trợ chẩn đoán hình ảnh, từ đó hiểu rõ nguyên lý nhận diện tổn thương và nâng cao mức độ tin cậy khi sử dụng công nghệ để giảm thiểu tỷ lệ bỏ sót 20-47% polyp trong quy trình tầm soát ung thư.
  • Kỹ sư phát triển hệ thống Thị giác máy tính và Nhận diện sinh trắc học: Các kỹ sư công nghệ có thể kế thừa trực tiếp mã nguồn kiến trúc ResCUNet và quy trình trích xuất vector đặc trưng 512 chiều từ bản đồ UV map để nâng cấp các hệ thống camera giám sát thông minh tại sân bay, tòa nhà và đô thị.
  • Doanh nghiệp phát triển giải pháp công nghệ y tế (MedTech): Đội ngũ nghiên cứu và phát triển (R&D) tại các công ty thiết bị y tế có thể khai thác khung kiến trúc Attention ResCUNeSt để tích hợp trực tiếp vào phần mềm máy nội soi tiêu chuẩn, nâng cao giá trị thương mại và năng lực cạnh tranh của sản phẩm.

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

1. Tại sao kiến trúc U-Net truyền thống lại gặp khó khăn khi phân đoạn các polyp có kích thước dưới 10 mm?
Kiến trúc U-Net thuần túy gặp hiện tượng suy giảm thông tin chi tiết qua các tầng tích chập sâu và khoảng cách ngữ nghĩa lớn giữa nhánh mã hóa và nhánh giải mã. Đối với các polyp nhỏ dưới 10 mm có độ tương phản mờ nhạt với niêm mạc, U-Net thông thường dễ bị bão hòa tín hiệu, dẫn đến việc bỏ sót hoặc phân đoạn sai lệch ranh giới tổn thương.

2. Cơ chế Split-Attention trong ResNeSt mang lại ưu thế gì vượt trội so với ResNet tiêu chuẩn?
ResNeSt phân chia các kênh đặc trưng thành nhóm cơ sở kết hợp cơ chế chú ý theo kênh cho từng nhánh mạng nhỏ. Điều này cho phép mô hình thu nhận thông tin ngữ cảnh đa tỉ lệ một cách linh hoạt, tạo ra các biểu diễn đặc trưng phong phú hơn ResNet truyền thống mà không làm bùng nổ số lượng tham số tính toán.

3. Mô hình Attention ResCUNet-GAN bảo toàn danh tính khuôn mặt bằng cách nào khi thực hiện xoay góc chụp?
Mô hình tích hợp mô-đun Identity Preserving sử dụng mạng FaceNet (được tiền huấn luyện trên 3,31 triệu ảnh từ 9.131 danh tính thuộc tập VGGFace2). Mô-đun này tính toán khoảng cách Euclidean giữa vector embedding 512 chiều của ảnh gốc và ảnh sinh ra, buộc mạng phải duy trì các đặc trưng nhân trắc học cốt lõi của khuôn mặt.

4. Vì sao hàm mất mát Tversky Loss lại vượt trội hơn Binary Cross-Entropy trong bài toán phân đoạn ảnh y tế?
Trong ảnh nội soi, vùng polyp chỉ chiếm tỷ lệ pixel rất nhỏ so với nền niêm mạc đại tràng. Hàm mất mát Tversky Loss cho phép gán trọng số phạt riêng biệt cho các trường hợp âm tính giả (False Negatives) và dương tính giả (False Positives), qua đó khắc phục triệt để hiện tượng mất cân bằng dữ liệu và tối ưu hóa chỉ số mDice.

5. Mô hình phân đoạn trong luận văn đã được kiểm thử độ ổn định xuyên trung tâm y tế như thế nào?
Nghiên cứu đã thiết lập 6 kịch bản kiểm thử độc lập, trong đó mô hình được huấn luyện trên tập dữ liệu CVC-Colon nhưng kiểm tra trực tiếp trên tập ETIS-Larib từ cơ sở y tế khác. Kết quả cho thấy mô hình vẫn duy trì độ chính xác cao và không bị hiện tượng quá khớp (overfitting).

Kết luận

Luận văn thạc sĩ "An Improved UNets Architecture and Its Applications" của tác giả Trần Quang Chung đã giải quyết xuất sắc các thách thức lớn trong thị giác máy tính thông qua các đóng góp khoa học nổi bật:

  • Đề xuất thành công ba kiến trúc nơ-ron tích chập cải tiến: ResCUNet, Attention ResCUNet và Attention ResCUNeSt, kết hợp đột phá giữa cơ chế cổng chú ý và kiến trúc mạng ghép đôi.
  • Xây dựng giải pháp tái tạo khuôn mặt trực diện từ ảnh góc nghiêng với chất lượng chân thực cao, bảo toàn nguyên vẹn danh tính trên không gian vector 512 chiều.
  • Đạt hiệu năng phân vùng polyp đường tiêu hóa vượt trội trên các tập chuẩn quốc tế Kvasir-SEG và CVC-ClinicDB, hỗ trợ thiết thực cho việc sàng lọc ung thư đại trực tràng.
  • Đóng góp phương pháp tối ưu hóa mạng nơ-ron sâu thông qua việc kết hợp cơ chế Split-Attention, hàm mất mát Tversky và các nhánh tổn thất phụ trợ Auxiliary Loss.
  • Cung cấp cơ sở lý thuyết vững chắc và quy trình thực nghiệm chuẩn mực, đã được công bố tại các diễn đàn khoa học uy tín trong nước và quốc tế (MAPR, SCI-Q1).

Trong giai đoạn 12 đến 24 tháng tới, hướng nghiên cứu tiếp theo sẽ tập trung vào việc tối ưu hóa tốc độ suy luận, lượng tử hóa mô hình để triển khai trên các vi xử lý nhúng và mở rộng thử nghiệm lâm sàng thực tế. Các viện nghiên cứu, doanh nghiệp công nghệ y tế và chuyên gia thị giác máy tính được khuyến khích tiếp cận, ứng dụng và phát triển mở rộng khung kiến trúc này để thúc đẩy chuyển đổi số y tế và công nghệ nhận diện thông minh.