Tổng quan nghiên cứu

Trong kỷ nguyên chuyển đổi số và công nghệ đa phương tiện, xử lý tín hiệu âm thanh kỹ thuật số với tần số lấy mẫu chuẩn 44.100 Hz (44,1 kHz) đóng vai trò then chốt trong việc tái tạo và nâng cao trải nghiệm thính giác. Bài toán tách nguồn âm thanh (Audio Source Separation), đặc biệt là phân tách nguồn âm nhạc (Music Source Separation), luôn đối mặt với thách thức lớn từ hiệu ứng "bữa tiệc cocktail" (cocktail party effect), hiện tượng rò rỉ quang phổ và sự suy giảm chất lượng âm thanh trải dài trên dải tần từ 20 Hz đến 20.000 Hz.

Đề tài luận văn thạc sĩ chuyên ngành Khoa học Máy tính "Tách nguồn âm thanh dựa trên tiếp cận học máy" do tác giả Thẩm Quốc Dũng thực hiện dưới sự hướng dẫn của PGS. Huỳnh Tường Nguyên và TS. Nguyễn Đức Dũng tại Trường Đại học Bách Khoa – Đại học Quốc gia TP. Hồ Chí Minh (hoàn thành tháng 12 năm 2023, bảo vệ tháng 01 năm 2024) tập trung giải quyết bài toán cốt lõi này. Mục tiêu nghiên cứu là khảo sát toàn diện các phương pháp học sâu hiện đại, phân tích hạn chế của các kiến trúc tiền nhiệm và đề xuất mô hình mạng nơ-ron tối ưu nhằm bóc tách 4 nguồn âm thanh mục tiêu (vocals, drums, bass, others) từ một bản phối tổng hợp stereo 2 kênh.

Phạm vi thực nghiệm của đề tài được triển khai trên tập dữ liệu chuẩn MusDB18-HQ với hơn 10 giờ âm thanh chất lượng cao gồm 150 bài hát hoàn chỉnh. Ý nghĩa khoa học và thực tiễn của công trình thể hiện qua việc giảm thiểu độ trễ xử lý xuống dưới 0,29 giây cho mỗi giây âm thanh, nâng cao chỉ số chất lượng nguồn trên độ méo dạng (Source-to-Distortion Ratio - SDR) vượt mốc 7,0 dB, mở ra hướng ứng dụng trực tiếp cho các hệ thống nhận diện giọng nói tự động (ASR), hội nghị trực tuyến (VoIP), truy xuất thông tin âm nhạc (MIR) và sản xuất âm thanh chuyên nghiệp.

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 trên nền tảng vững chắc của lý thuyết xử lý tín hiệu số (DSP) và các mô hình học sâu hiện đại:

  • Lý thuyết lấy mẫu và biểu diễn tín hiệu: Tuân thủ định lý Nyquist-Shannon với tần số lấy mẫu tối thiểu đạt $F_s \ge 2f_{max}$, đảm bảo tái tạo chính xác tín hiệu âm thanh 16-bit với dải động xấp xỉ 96,32 dB mà không bị hiện tượng chồng phổ (alias).
  • Mô hình biểu diễn thời gian - tần số: Sử dụng phép biến đổi Fourier ngắn hạn (Short-Time Fourier Transform - STFT) kết hợp hàm cửa sổ Hann với các khung phân tích chồng lấp (overlapping frames) để chuyển đổi sóng âm miền thời gian thành biểu đồ quang phổ (spectrogram) hai chiều.
  • Khung số phức làm kênh tín hiệu (Complex as Channel - CaC): Biểu diễn ma trận phổ phức dưới dạng tensor thực đa kênh kích thước $2c \times T \times F$, giữ trọn vẹn cả thông tin biên độ (magnitude) lẫn góc pha dao động ban đầu ($\phi \in [-\pi, \pi]$).
  • Các khái niệm cốt lõi bổ trợ: Mặt nạ nhị phân lý tưởng (IBM), mặt nạ tỷ lệ lý tưởng (IRM), mặt nạ nhạy pha (PSM), bộ lọc Wiener đa kênh (Multi-channel Wiener Filter - MWF), và vector đặc trưng 39 chiều Mel-Frequency Cepstral Coefficients (MFCC).

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

  • Nguồn dữ liệu và cỡ mẫu: Dữ liệu thực nghiệm chính là tập MusDB18-HQ gồm 150 bài hát stereo (100 bài dùng cho huấn luyện và 50 bài dùng cho kiểm thử) được lưu trữ dưới định dạng WAV không nén với tần số 44,1 kHz. Ngoài ra, nghiên cứu tham chiếu tập WSJ0-mix với hơn 30 giờ giọng nói tổng hợp chịu mức nhiễu ngẫu nhiên SNR từ 0 đến 5 dB và tập WHAMR! chứa tạp âm môi trường và độ vang phòng.
  • Phương pháp chọn mẫu: Chọn mẫu phân tầng ngẫu nhiên theo cấu trúc chuẩn của cuộc thi SiSEC 2018 và Music Demixing Challenge (MDX), bảo đảm tính đa dạng về thể loại âm nhạc, tiết tấu và phổ tần số.
  • Lý do lựa chọn phương pháp phân tích: Thay vì tiếp cận theo hướng huấn luyện 4 mô hình độc lập cho từng nhạc cụ (gây tốn kém gấp 4 lần tài nguyên GPU), nghiên cứu lựa chọn phương pháp học có giám sát đa mục tiêu (multi-target learning). Mô hình tích hợp khối Time-Frequency Convolutions (TFC), mạng nơ-ron phân tán thời gian (TDF) và kiến trúc Conformer hai giai đoạn (Two-stages Conformer) nhằm khai thác đồng thời tương quan cục bộ và ngữ cảnh toàn cục của âm sắc.
  • Timeline nghiên cứu: Toàn bộ quy trình từ khảo sát lý thuyết, tiền xử lý dữ liệu, thiết kế kiến trúc eCMU, huấn luyện mô hình trên GPU Tesla T4 đến đánh giá kết quả được thực hiện nghiêm ngặt từ ngày 06/02/2023 đến ngày 10/12/2023.

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 và kiểm thử trên 50 bài hát của tập MusDB18-HQ đã mang lại những phát hiện quan trọng:

  1. Hiệu năng và độ phức tạp của Open-Unmix (UMX): Mô hình UMX sử dụng 3 lớp Bi-LSTM với 8,93 triệu tham số đạt điểm SDR tổng thể 5,33 dB. Thời gian suy diễn đạt 0,13 giây cho mỗi 1 giây âm thanh. Tuy nhiên, mô hình bị mất mát nghiêm trọng các dải tần số cao trên 10 kHz khi xử lý giọng hát và nhạc cụ bộ dây.
  2. Ưu thế tốc độ của Spleeter: Spleeter với kiến trúc U-Net 12 lớp (6 encoder, 6 decoder) đạt SDR 5,90 dB và tốc độ suy diễn nhanh vượt trội là 0,08 giây/giây âm thanh. Điểm hạn chế là mô hình xuất hiện hiện tượng rò rỉ quang phổ (spectral leakage) ở các đoạn im lặng và phụ thuộc vào tập dữ liệu ngoài lên đến 25.000 bài hát.
  3. Chất lượng vượt trội nhưng tốn kém tài nguyên của Demucs v3: Demucs v3 trên miền sóng âm đạt SDR lên đến 7,70 dB, tạo ra chất lượng âm thanh trong trẻo. Tuy nhiên, mô hình sở hữu tới 133,75 triệu tham số, đòi hỏi 360 epoch huấn luyện trên cụm 8 GPU NVIDIA V100 16GB (khoảng 4 giờ/epoch), gây tắc nghẽn khi triển khai trên phần cứng giới hạn.
  4. Hiệu quả của kiến trúc eCMU và Two-stages Conformer: Phương pháp đề xuất eCMU kết hợp khối D2-TDF duy trì được chất lượng tách âm tương đương các mô hình SOTA, giảm hơn 40% kích thước tham số và triệt tiêu đáng kể hiện tượng nhiễu nhân tạo (artifacts) ở tần số cao.

Thảo luận kết quả

Khi phân tích kết quả qua bảng thống kê hiệu năng và biểu đồ nhiệt quang phổ (spectrogram heatmap), nguyên nhân của sự khác biệt giữa các mô hình trở nên rõ ràng. Các mô hình hoạt động trên miền biên độ đơn thuần (như UMX) thường bỏ qua pha dao động, dẫn đến âm thanh tái tạo bị mờ nhạt và mất chi tiết âm sắc. Ngược lại, phương pháp xử lý số phức CaC kết hợp bộ lọc Wiener đa kênh (MWF) giúp bảo toàn cấu trúc hài âm (harmonics) của nốt nhạc.

Biểu đồ phân bố năng lượng theo dải tần cho thấy hiện tượng nhiễu tần số cao của Demucs xuất phát từ việc tích chập trực tiếp trên dạng sóng rời rạc không có cơ chế lọc thông dải cục bộ. Trong khi đó, kiến trúc Two-stages Conformer kết hợp ResUNet giải quyết triệt để lỗi đánh nhãn sai lệch (label noise) trong tập dữ liệu (điển hình như trường hợp bài hát "Timboz - Pony" bị mất kênh vocal trong bản gốc), giúp âm thanh đầu ra đạt điểm số đánh giá chủ quan MOS ước tính trên 4,2/5,0.

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

Dựa trên các phân tích thực nghiệm, nghiên cứu đề xuất 4 nhóm giải pháp kỹ thuật cụ thể:

  1. Chuẩn hóa kiến trúc mô hình chia sẻ trọng số (Multi-target Shared Backbone): Chuyển đổi toàn bộ quy trình tách 4 luồng riêng biệt sang mô hình thống nhất duy nhất. Chỉ số mục tiêu: Giảm dung lượng lưu trữ mô hình xuống dưới 20 triệu tham số, rút ngắn thời gian huấn luyện 60%. Thời gian: Quý 1/2024 - Quý 2/2024. Chủ thể thực hiện: Nhóm nghiên cứu AI và kỹ sư xử lý âm thanh.
  2. Tích hợp khối nhận biết pha eCMU và hậu xử lý Wiener đa kênh (MWF): Triển khai toán tử ước tính ma trận hiệp phương sai không gian (SCM) và mật độ phổ công suất (PSD) ở bước suy diễn. Chỉ số mục tiêu: Tăng tỷ số nguồn trên can nhiễu (SIR) thêm 1,5 đến 2,0 dB, loại bỏ 95% nhiễu rò rỉ ở dải tần số trên 15 kHz. Thời gian: Quý 3/2024. Chủ thể thực hiện: Đội ngũ phát triển thuật toán DSP.
  3. Lọc sạch và làm giàu tập dữ liệu huấn luyện bằng VAD tự động: Xây dựng module tự động phát hiện hoạt tính giọng nói (Voice Activity Detection - VAD) để khử nhiễu gán nhãn trên 150 bài hát của MusDB18-HQ. Chỉ số mục tiêu: Loại bỏ 100% các đoạn im lặng bị gán nhãn sai. Thời gian: Quý 4/2024. Chủ thể thực hiện: Kỹ sư dữ liệu âm thanh.
  4. Lượng tử hóa mô hình và triển khai trên thiết bị biên (Edge AI): Áp dụng kỹ thuật nén mô hình TensorRT, lượng tử hóa INT8/FP16 cho kiến trúc Conformer. Chỉ số mục tiêu: Đạt thời gian suy diễn thực tế dưới 0,05 giây/giây âm thanh trên CPU máy tính cá nhân. Thời gian: Năm 2025. Chủ thể thực hiện: Các doanh nghiệp phát triển ứng dụng di động, karaoke và viễn thông VoIP.

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

Công trình nghiên cứu mang giá trị thực tiễn cao cho 4 nhóm đối tượng chính:

  • Kỹ sư và nhà nghiên cứu Trí tuệ nhân tạo (AI/ML Engineers): Tiếp cận phương pháp hiện thực mạng Conformer, U-Net cải tiến và kỹ thuật xử lý tín hiệu số phức CaC, phục vụ phát triển các mô hình học sâu cho âm thanh trên hạ tầng phần cứng GPU Tesla đơn lẻ.
  • Nhà phát triển sản phẩm công nghệ âm nhạc (Music Tech Developers): Ứng dụng thuật toán để bóc tách bài hát thành 4 stems (vocals, trống, bass, nhạc cụ khác) phục vụ tính năng tách lời làm nhạc nền karaoke, căn chỉnh lời bài hát tự động (lyrics alignment) và sản xuất âm nhạc thông minh.
  • Doanh nghiệp viễn thông và nền tảng đàm thoại trực tuyến (VoIP & Telecom): Sử dụng các cơ chế lọc nhiễu cocktail party và tăng cường chất lượng âm thanh để nâng cao độ rõ nét của hội nghị trực tuyến, đồng thời cải thiện độ chính xác của hệ thống nhận diện giọng nói tự động ASR từ 15% đến 25%.
  • Học viên cao học và sinh viên chuyên ngành Khoa học Máy tính: Sử dụng luận văn như tài liệu học thuật chuẩn mực về lý thuyết xử lý tín hiệu âm thanh (từ sóng cơ học, thang đo Mel, MFCC đến biến đổi Fourier) và phương pháp luận nghiên cứu khoa học thực nghiệm.

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

1. Phép biến đổi Fourier ngắn hạn (STFT) đóng vai trò gì trong bài toán tách nguồn âm thanh?
STFT chia tín hiệu âm thanh liên tục thành các khung ngắn (frames) có kích thước cố định và áp dụng hàm cửa sổ Hann để tính toán phân bố tần số theo thời gian. Kỹ thuật này giúp chuyển đổi sóng âm 1 chiều thành biểu đồ quang phổ 2 chiều, cho phép mạng nơ-ron học các mẫu âm sắc đặc trưng trên từng dải tần số cụ thể.

2. Vì sao mô hình Demucs đạt chỉ số SDR cao (7,7 dB) nhưng lại khó triển khai thực tế?
Demucs xử lý trực tiếp trên dạng sóng thời gian với 133,75 triệu tham số, đòi hỏi 360 epoch huấn luyện trên cụm 8 GPU V100 16GB và có độ trễ suy diễn 0,29 giây/giây âm thanh. Chi phí phần cứng và thời gian tính toán lớn khiến Demucs khó tích hợp vào các thiết bị di động hoặc ứng dụng thời gian thực.

3. Khung Complex as Channel (CaC) mang lại ưu thế gì so với xử lý Spectrogram truyền thống?
Xử lý Spectrogram truyền thống chỉ quan tâm đến biên độ và loại bỏ thông tin pha dao động, khiến âm thanh tái tạo bị méo dạng. CaC xem phần thực và phần ảo của phổ Fourier phức như các kênh độc lập ($2c$ kênh), giúp mô hình bảo toàn trọn vẹn thông tin pha và khôi phục tín hiệu gốc chính xác hơn.

4. Sự khác biệt giữa hai chuẩn đo SDR và MOS trong đánh giá chất lượng âm thanh là gì?
SDR (Source-to-Distortion Ratio) là thước đo khách quan tính bằng decibel (dB) biểu diễn tỷ lệ giữa năng lượng nguồn sạch và tổng sai số do can nhiễu, tạp âm và méo thuật toán. Trong khi đó, MOS (Mean Opinion Score) là thang đo chủ quan từ 1 đến 5 điểm do con người trực tiếp lắng nghe và đánh giá độ tự nhiên của âm thanh.

5. Phương pháp eCMU và kiến trúc đa mục tiêu trong luận văn giải quyết bài toán gì?
Phương pháp này loại bỏ sự cồng kềnh của việc duy trì 4 mô hình riêng biệt cho 4 nguồn âm thanh, gom toàn bộ quá trình tách âm vào một kiến trúc duy nhất chia sẻ trọng số. Giải pháp này giúp giảm hơn 40% chi phí tính toán, tận dụng mối tương quan giữa các nhạc cụ và triệt tiêu hiện tượng rò rỉ âm thanh.

Kết luận

Luận văn thạc sĩ "Tách nguồn âm thanh dựa trên tiếp cận học máy" của tác giả Thẩm Quốc Dũng đã hoàn thành xuất sắc các mục tiêu nghiên cứu với những đóng góp nổi bật:

  • Hệ thống hóa toàn diện cơ sở lý thuyết xử lý tín hiệu âm thanh số từ sóng cơ học, thang Mel, MFCC đến các phép biến đổi Fourier phức.
  • Khảo sát thực nghiệm chi tiết các mô hình SOTA gồm Open-Unmix (8,93M tham số), Spleeter (9,87M tham số) và Demucs v3 (133,75M tham số, SDR 7,7 dB) trên tập dữ liệu chuẩn MusDB18-HQ.
  • Phân tích nguyên nhân cốt lõi gây suy giảm chất lượng âm thanh: hiện tượng mất pha, rò rỉ quang phổ ở đoạn im lặng và nhiễu tần số cao.
  • Đề xuất giải pháp kiến trúc đa mục tiêu eCMU kết hợp Two-stages Conformer và D2-TDF, tối ưu hóa độ chính xác và giảm hơn 40% chi phí tính toán so với các mô hình độc lập.
  • Định hình lộ trình phát triển rõ ràng: ứng dụng lượng tử hóa mô hình và triển khai trên các hệ thống hội thoại VoIP, trợ lý ảo ASR và ứng dụng giải trí âm nhạc trong giai đoạn 2024–2025.

Quý độc giả, nhà nghiên cứu và kỹ sư công nghệ quan tâm đến giải pháp xử lý âm thanh số có thể tham khảo toàn văn luận văn tại Thư viện Trường Đại học Bách Khoa – ĐHQG-HCM để ứng dụng và phát triển các mô hình học sâu tách nguồn âm thanh tiên tiến trong tương lai.