Bối cảnh và vấn đề nghiên cứu

Trong bối cảnh bùng nổ dữ liệu số và sự phát triển của năng lực tính toán hiện đại, học sâu (Deep Learning) thuộc lĩnh vực Trí tuệ Nhân tạo (AI) đã khẳng định vai trò then chốt thông qua việc sử dụng các mạng nơ-ron nhân tạo (ANNs) để mô hình hóa và học các mẫu dữ liệu phức tạp. Mặc dù học sâu đã đạt được nhiều thành tựu trong các bài toán nhận dạng hình ảnh, xử lý ngôn ngữ tự nhiên, tự lái ô tô và y sinh học, việc xây dựng và huấn luyện mô hình vẫn đối mặt với thách thức lớn về chi phí tính toán, thời gian huấn luyện và hiện tượng quá khớp (overfitting). Trong đó, việc lựa chọn thuật toán tối ưu đóng vai trò quyết định đến tốc độ hội tụ, tính ổn định và khả năng tổng quát hóa của mô hình trên dữ liệu mới.

Nhằm giải quyết vấn đề trên, đề tài tập trung nghiên cứu, thực nghiệm và so sánh hiệu quả của các thuật toán tối ưu phổ biến gồm Stochastic Gradient Descent (SGD), SGD with Momentum, RMSProp và Adam khi huấn luyện mô hình mạng nơ-ron phân loại ảnh chữ số viết tay trên bộ dữ liệu MNIST, kết hợp với các kỹ thuật điều chuẩn (regularization) là L2 Regularization và Dropout.

Nhóm tác giả xác định 4 mục tiêu nghiên cứu cụ thể như sau:

  1. Mục tiêu 1: So sánh hiệu quả của các thuật toán tối ưu SGD, Momentum, RMSProp và Adam trong việc huấn luyện mô hình mạng nơ-ron cho bài toán phân loại ảnh MNIST.
  2. Mục tiêu 2: Áp dụng kỹ thuật L2 Regularization và Dropout vào mô hình để kiểm soát hiện tượng quá khớp (overfitting) và cải thiện khả năng tổng quát hóa của mạng nơ-ron.
  3. Mục tiêu 3: Đánh giá hiệu năng của mỗi thuật toán tối ưu dựa trên các tiêu chí định lượng gồm độ chính xác (Accuracy), giá trị hàm mất mát (Loss) trên tập kiểm tra và thời gian huấn luyện.
  4. Mục tiêu 4: Đưa ra các kết luận và đề xuất dựa trên kết quả thực nghiệm nhằm hỗ trợ việc lựa chọn thuật toán tối ưu phù hợp cho từng bài toán học sâu cụ thể.
  • Đối tượng nghiên cứu: Các thuật toán tối ưu hóa (SGD, Momentum, RMSProp, Adam), các kỹ thuật chống quá khớp (L2 Regularization, Dropout) và mô hình mạng nơ-ron phân loại ảnh.
  • Phạm vi nghiên cứu: Bài toán phân loại ảnh chữ số viết tay từ 0 đến 9 trên bộ cơ sở dữ liệu tiêu chuẩn MNIST; môi trường thực nghiệm lập trình bằng Python với các thư viện TensorFlow và Keras.

Cơ sở lý thuyết và phương pháp

Khung lý thuyết và mô hình

Nghiên cứu dựa trên nền tảng lý thuyết mạng nơ-ron nhân tạo và học sâu, bao gồm các cấu trúc và khái niệm cốt lõi:

  • Kiến trúc mạng nơ-ron: Mạng nơ-ron truyền thẳng (Feedforward Neural Network) và Mạng nơ-ron tích chập (Convolutional Neural Network - CNN) với các tầng chức năng:
    • Lớp tích chập (Convolutional Layer): Trích xuất đặc trưng không gian thông qua các bộ lọc (kernel), bước nhảy (stride), phần đệm (padding) và nhóm bộ lọc (filter group).
    • Lớp lấy mẫu xuống (Pooling/Subsampling Layer): Giảm kích thước không gian và số lượng tham số thông qua Max Pooling hoặc Average Pooling.
    • Lớp kết nối đầy đủ (Fully-Connected Layer - FC): Tổng hợp các đặc trưng đã được làm phẳng (flatten) để đưa ra dự đoán ở tầng đầu ra thông qua quá trình lan truyền xuôi (Forward Propagation) và lan truyền ngược (Backpropagation).
  • Hàm kích hoạt (Activation Functions): Tạo tính phi tuyến tính cho mạng, bao gồm Sigmoid, Tanh, ReLU ($f(x) = \max(0, x)$), Leaky ReLU và Softmax (chuyển đổi đầu ra thành phân phối xác suất cho bài toán phân loại đa lớp).
  • Hàm mất mát (Loss Function): Sử dụng hàm Binary Cross Entropy cho bài toán nhị phân và Multiclass Cross Entropy cho bài toán phân loại đa lớp: $$\text{Multiclass Cross Entropy} = -\sum_{i=1}^{n} y_i \log(\hat{y}_i)$$ Trong đó $n$ là số lớp, $y_i$ là nhãn thực tế dạng one-hot encoding, và $\hat{y}_i$ là xác suất dự đoán của mô hình.
  • Kỹ thuật chống quá khớp (Regularization):
    • L2 Regularization (Ridge): Bổ sung số hạng phạt tỉ lệ với tổng bình phương trọng số $\lambda \sum m_i^2$ vào hàm mất mát để hạn chế độ lớn tham số.
    • Dropout: Ngẫu nhiên loại bỏ một tỉ lệ nơ-ron với xác suất $p$ trong mỗi chu kỳ huấn luyện nhằm tránh sự phụ thuộc lẫn nhau giữa các nút mạng.
    • Các kỹ thuật tham khảo khác: Data Augmentation (xoay, lật, phóng to, cắt ảnh), DropConnect và Early Stopping.

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

Đề tài sử dụng phương pháp nghiên cứu thực nghiệm định lượng kết hợp khảo sát lý thuyết:

  • Tổng quan các công trình nghiên cứu tiêu biểu của Ruder (2016), Reddi và cộng sự (2018), Li và cộng sự (2018).
  • Thiết kế mô hình mạng nơ-ron đồng nhất, cố định các siêu tham số cơ bản và lần lượt thay đổi các thuật toán tối ưu hóa (SGD, Momentum, RMSProp, Adam) kết hợp với kỹ thuật L2 Regularization và Dropout.
  • Thu thập và đo lường các chỉ số thực nghiệm trên tập huấn luyện và tập kiểm tra.

Nguồn dữ liệu

Nghiên cứu sử dụng bộ dữ liệu chuẩn MNIST (LeCun et al., 1998) bao gồm 70.000 ảnh xám chữ số viết tay kích thước $28 \times 28$ pixel (tương ứng không gian đầu vào 784 chiều), được gán nhãn thành 10 lớp (từ 0 đến 9). Bộ dữ liệu gồm 4 tệp nhị phân nén:

  • train-images-idx3-ubyte.gz: Dữ liệu 60.000 ảnh tập huấn luyện.
  • train-labels-idx1-ubyte.gz: Nhãn của 60.000 mẫu tập huấn luyện.
  • t10k-images-idx3-ubyte.gz: Dữ liệu 10.000 ảnh tập kiểm tra.
  • t10k-labels-idx1-ubyte.gz: Nhãn của 10.000 mẫu tập kiểm tra.

Thiết kế và triển khai thực nghiệm

Kiến trúc mô hình mạng nơ-ron

Mô hình thực nghiệm được xây dựng dưới dạng mạng nơ-ron truyền thẳng (Feedforward Neural Network) với các tham số cấu hình:

  • Kích thước đầu vào (Input Dimension): $28 \times 28 = 784$ nơ-ron (sau khi làm phẳng ma trận ảnh).
  • Lớp ẩn 1 (Hidden Layer 1): 128 nơ-ron, hàm kích hoạt ReLU.
  • Lớp ẩn 2 (Hidden Layer 2): 64 nơ-ron, hàm kích hoạt ReLU.
  • Lớp đầu ra (Output Layer): 10 nơ-ron (tương ứng 10 chữ số), hàm kích hoạt Softmax.
  • Kích thước lô nhỏ (Minibatch size): $m = 100$.

Nguyên lý và công thức các thuật toán tối ưu hóa triển khai

Thuật toán Cơ chế cập nhật tham số Đặc điểm chính
SGD $\theta_{t+1} = \theta_t - \eta \nabla_\theta J(\theta_t)$ Cập nhật trọng số theo từng mẫu ngẫu nhiên hoặc mini-batch; chi phí tính toán thấp nhưng đường đi dao động dạng zig-zag.
Momentum $v_t = \gamma v_{t-1} + \eta \nabla_\theta J(\theta)$
$\theta = \theta - v_t$
Tích lũy vector gradient quá khứ với hệ số $\gamma \approx 0.9$ giúp giảm dao động và tăng tốc độ vượt qua các điểm cực tiểu cục bộ.
RMSProp $E[g^2]t = \alpha E[g^2]{t-1} + (1-\alpha)g_t^2$
$\theta_{t+1} = \theta_t - \frac{\eta}{\sqrt{E[g^2]_t + \epsilon}} g_t$
Sử dụng trung bình động bình phương gradient để tự động điều chỉnh tốc độ học thích ứng cho từng tham số; giá trị mặc định $\alpha = 0.9$, $\eta = 0.001$.
Adam $m_t = \beta_1 m_{t-1} + (1-\beta_1)g_t$
$v_t = \beta_2 v_{t-1} + (1-\beta_2)g_t^2$
$\hat{m}_t = \frac{m_t}{1-\beta_1^t}, \quad \hat{v}t = \frac{v_t}{1-\beta_2^t}$
$\theta
{t+1} = \theta_t - \frac{\eta}{\sqrt{\hat{v}_t} + \epsilon} \hat{m}_t$
Kết hợp ước lượng moment bậc một ($m_t$) và moment bậc hai ($v_t$) kèm hiệu chỉnh sai lệch thiên vị (bias correction); siêu tham số đề xuất: $\beta_1 = 0.9$, $\beta_2 = 0.999$, $\epsilon = 10^{-8}$.

Môi trường công nghệ

Quy trình thực nghiệm được triển khai trên nền tảng ngôn ngữ Python kết hợp các thư viện chuyên dụng:

  • tensorflow và tensorflow.keras: Khởi tạo mô hình tuần tự (Sequential), xây dựng các tầng (Dense, Dropout, Activation), cấu hình thuật toán tối ưu (SGD, RMSprop, Adam) và bộ điều chuẩn (regularizers.l2).
  • tensorflow.datasets và idx2numpy: Tải và chuyển đổi tập dữ liệu MNIST từ định dạng byte sang mảng NumPy.
  • numpy: Xử lý mảng số học và tính toán ma trận.
  • matplotlib.pyplot và ipywidgets: Trực quan hóa dữ liệu đầu vào, biểu đồ hàm mất mát và các thành phần tương tác.
  • tensorflow.utils: Sử dụng hàm to_categorical để mã hóa nhãn sang dạng one-hot và plot_model để xuất sơ đồ kiến trúc mạng.

Nội dung chính theo từng chương

Chương 1: Giới Thiệu

Chương 1 trình bày tổng quan về lĩnh vực học sâu và mạng nơ-ron nhân tạo, nêu bật vai trò của các framework mã nguồn mở phổ biến như TensorFlow, PyTorch, Keras, MXNet và Caffe. Tác giả phân tích các nhánh ứng dụng chính của học sâu trong thực tiễn gồm nhận dạng hình ảnh, xử lý ngôn ngữ tự nhiên (NLP), xe tự lái và y sinh học. Đồng thời, chương này xác lập 4 mục tiêu nghiên cứu cụ thể, phạm vi đề tài và các tiêu chí đánh giá hiệu năng mô hình (Accuracy, Loss, Training Time).

Chương 2: Khảo Sát Tài Liệu

Chương 2 tổng hợp các công trình nghiên cứu nền tảng về tối ưu hóa trong học sâu của Ruder (2016), Reddi và cộng sự (2018) và Li và cộng sự (2018). Tác giả phân tích chi tiết cấu trúc mạng nơ-ron tích chập (CNN) gồm lớp tích chập, lớp lấy mẫu xuống (Max Pooling, Average Pooling), lớp kết nối đầy đủ (FC) cùng cơ chế lan truyền xuôi và lan truyền ngược. Tiếp đó, chương khảo sát đặc tính toán học của các hàm kích hoạt (Sigmoid, Tanh, ReLU, Leaky ReLU, Softmax) và nguyên lý của hàm mất mát Cross Entropy cùng tham số tốc độ học (learning rate).

Chương 3: Phương Pháp Nghiên Cứu

Chương 3 đi sâu vào bản chất toán học và thuật toán của 4 bộ tối ưu hóa: SGD, SGD with Momentum, RMSProp và Adam, phân tích rõ cơ chế "Momentum steering" và cách thích ứng tốc độ học theo từng chiều không gian tham số. Chương cũng làm rõ nguyên lý của các kỹ thuật kiểm soát quá khớp gồm Data Augmentation, L2 Regularization, Dropout (cách chọn hệ số $p \approx 0.5$ cho lớp ẩn và $p = 1.0$ cho lớp đầu vào), DropConnect và Early Stopping. Cuối cùng, chương thiết lập hệ thống công thức cho các chỉ số đánh giá gồm Accuracy, Precision, Recall, F1-score, Confusion Matrix và đường cong ROC/AUC.

Chương 4: Chuẩn Bị Dữ Liệu

Chương 4 giới thiệu chi tiết về bộ cơ sở dữ liệu MNIST với 70.000 mẫu ảnh chữ số viết tay kích thước $28 \times 28$ pixel, phân tích cấu trúc 4 tệp dữ liệu nén nhị phân (.gz) và phân chia tập huấn luyện (60.000 ảnh) cùng tập kiểm tra (10.000 ảnh). Tác giả làm rõ lý do lựa chọn MNIST dựa trên 3 tiêu chí: độ phổ biến cao dễ đối sánh, khả năng tiếp cận tích hợp sẵn trong TensorFlow/Keras, và kích thước vừa phải giúp tối ưu hóa thời gian và tài nguyên tính toán thực nghiệm.

Chương 5: Triển Khai Và Huấn Luyện Mô Hình

Chương 5 trình bày toàn bộ quy trình kỹ thuật gồm tiền xử lý dữ liệu (chuyển đổi tệp IDX sang mảng NumPy, chuẩn hóa giá trị điểm ảnh, chuyển đổi nhãn sang one-hot encoding), trực quan hóa mẫu dữ liệu, thiết kế kiến trúc mạng nơ-ron 3 lớp ẩn (128-64-10) bằng Keras Sequential API. Chương mô tả quá trình biên dịch và huấn luyện mô hình với từng cấu hình thuật toán tối ưu kết hợp cùng các lớp Dropout và tham số l2 regularizer, ghi nhận biến thiên của hàm mất mát và độ chính xác qua từng epoch.

Chương 6: Đánh Giá Kết Luận

Chương 6 tổng kết và đối sánh hiệu năng tổng thể của các thuật toán tối ưu trên tập kiểm tra MNIST. Tác giả phân tích ưu điểm và hạn chế cố hữu của từng hàm tối ưu khi huấn luyện mô hình thực tế, đồng thời tổng hợp các đóng góp chính của đề tài và đề xuất các hướng phát triển nghiên cứu tiếp theo.


Kết quả và đóng góp

Hệ thống chỉ số đánh giá được thiết lập

Nghiên cứu xây dựng khung đánh giá hiệu năng toàn diện cho mô hình phân loại ảnh dựa trên các chỉ số định lượng:

Chỉ số Công thức xác định Mục đích đánh giá
Accuracy $\text{Accuracy} = \frac{\text{Số lượng dự đoán đúng}}{\text{Tổng số mẫu}}$ Đánh giá tỉ lệ phân loại chính xác tổng thể của mô hình trên toàn bộ tập dữ liệu.
Precision $\text{Precision} = \frac{TP}{TP + FP}$ Đo lường mức độ chính xác trong số các mẫu được mô hình dự đoán là dương tính.
Recall $\text{Recall} = \frac{TP}{TP + FN}$ Đo lường tỷ lệ các mẫu thực tế dương tính được mô hình nhận dạng thành công.
F1-score $\text{F1-score} = 2 \times \frac{\text{Precision} \times \text{Recall}}{\text{Precision} + \text{Recall}}$ Trung bình điều hòa giữa Precision và Recall khi cần cân bằng hai tiêu chí.
Loss $\text{Cross Entropy Loss}$ Đánh giá mức độ sai lệch giữa phân phối xác suất dự đoán và nhãn thực tế.
ROC & AUC Biểu đồ biến thiên giữa TPR và FPR Đánh giá năng lực phân loại tổng quát; giá trị AUC tiệm cận 1 biểu thị mô hình đạt hiệu năng cao.

Đóng góp của đề tài

  • Tổng hợp lý thuyết và công thức toán học có hệ thống: Đề tài làm rõ cơ chế tính toán gradient, quy tắc cập nhật tham số của SGD, Momentum, RMSProp và Adam, cùng các kỹ thuật điều chuẩn L2 và Dropout.
  • Xây dựng quy trình thực nghiệm chuẩn mực trên Keras/TensorFlow: Cung cấp cấu hình cụ thể cho mạng nơ-ron phân loại ảnh MNIST với kích thước minibatch 100, hàm kích hoạt ReLU và Softmax.
  • Đưa ra cơ sở hỗ trợ lựa chọn bộ tối ưu: Kết quả phân tích giúp xác định đặc tính hội tụ và mức độ ổn định của từng thuật toán, tạo tiền đề định hình chiến lược lựa chọn bộ tối ưu hóa phù hợp với yêu cầu bài toán.

Hạn chế và hướng nghiên cứu tiếp

Hạn chế được xác định trong văn bản

  • Hạn chế của thuật toán SGD: Quỹ đạo cập nhật tham số dao động mạnh (zig-zag) do gradient tính trên từng mẫu hoặc mini-batch không đại diện hoàn toàn cho tập dữ liệu; chưa tự giải quyết được vấn đề lựa chọn tốc độ học cố định và độ nhạy với điểm khởi tạo ban đầu.
  • Hạn chế của thuật toán Momentum: Phụ thuộc vào việc điều chỉnh siêu tham số hệ số động lượng $\gamma$; vẫn có khả năng rơi vào các điểm cực tiểu cục bộ nếu thiết lập không phù hợp.
  • Hạn chế của thuật toán RMSProp: Dù điều chỉnh tốt tốc độ học theo trung bình động bình phương gradient nhưng nghiệm hội tụ có thể dừng ở mức cực tiểu cục bộ (local minimum) thay vì cực tiểu toàn cục.
  • Phạm vi mô hình và dữ liệu: Thử nghiệm tập trung trên mô hình mạng truyền thẳng và tập dữ liệu chữ số viết tay MNIST với kích thước nhỏ ($28 \times 28$ pixel), chưa phản ánh toàn bộ các thách thức tính toán phức tạp trên các tập dữ liệu ảnh lớn hơn.

Hướng phát triển của đề tài

  • Mở rộng thử nghiệm so sánh các thuật toán tối ưu hóa trên các kiến trúc Mạng nơ-ron tích chập (CNN) sâu hơn.
  • Thực hiện đánh giá trên các bộ cơ sở dữ liệu hình ảnh đa chiều và phức tạp hơn như CIFAR-10 hoặc ImageNet.
  • Nghiên cứu kết hợp các kỹ thuật điều chỉnh tốc độ học động (learning rate scheduling) và các biến thể tối ưu hóa nâng cao khác.

Giá trị tham khảo

Báo cáo đồ án là tài liệu tham khảo hữu ích cho sinh viên, học viên thuộc các chuyên ngành Khoa học Dữ liệu, Trí tuệ Nhân tạo, Công nghệ Thông tin và Kỹ thuật Máy tính:

  • Phần phương pháp nghiên cứu (Chương 3): Cung cấp tài liệu tổng quan mạch lạc về biểu thức toán học, nguyên lý hoạt động của các thuật toán tối ưu (SGD, Momentum, RMSProp, Adam) và các phương pháp chống quá khớp (L2, Dropout, DropConnect, Early Stopping).
  • Phần chuẩn bị dữ liệu và triển khai (Chương 4 & 5): Hữu ích cho người làm đề tài thực hành nhập môn Deep Learning trong việc xử lý tệp nhị phân IDX của MNIST, chuyển đổi mảng NumPy, mã hóa nhãn to_categorical và thiết kế mô hình với TensorFlow/Keras API.

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

1. Đề tài khảo sát và so sánh những thuật toán tối ưu hóa nào?
Đề tài khảo sát và so sánh 4 thuật toán tối ưu hóa phổ biến trong học sâu gồm: Stochastic Gradient Descent (SGD), SGD với động lượng (SGD with Momentum), Root Mean Square Propagation (RMSProp) và Adaptive Moment Estimation (Adam).

2. Kiến trúc mạng nơ-ron thực nghiệm trong báo cáo được thiết kế như thế nào?
Mô hình là mạng nơ-ron truyền thẳng (Feedforward Neural Network) gồm đầu vào 784 chiều ($28 \times 28$), 3 lớp ẩn với số nơ-ron lần lượt là 128, 64 và 10 (lớp đầu ra). Các lớp ẩn sử dụng hàm kích hoạt ReLU, lớp đầu ra sử dụng hàm kích hoạt Softmax, và kích thước lô nhỏ (minibatch) được thiết lập là $m = 100$.

3. Kỹ thuật L2 Regularization và Dropout có vai trò gì trong mô hình?
Kỹ thuật L2 Regularization bổ sung số hạng phạt bình phương trọng số vào hàm mất mát để kiểm soát không cho các tham số tăng quá lớn. Dropout ngẫu nhiên loại bỏ một tỉ lệ nơ-ron (với xác suất $p \approx 0.5$ ở lớp ẩn) trong quá trình huấn luyện nhằm ngăn chặn hiện tượng quá khớp (overfitting) và tăng cường khả năng tổng quát hóa của mô hình.

4. Vì sao nhóm nghiên cứu lựa chọn tập dữ liệu MNIST cho bài toán thực nghiệm?
Bộ dữ liệu MNIST được lựa chọn do 3 yếu tố: tính phổ biến chuẩn mực giúp dễ so sánh kết quả với các nghiên cứu khác; khả năng tiếp cận thuận tiện qua các thư viện TensorFlow/Keras; và kích thước dữ liệu vừa phải (60.000 mẫu huấn luyện, 10.000 mẫu kiểm tra kích thước $28 \times 28$) giúp tiết kiệm thời gian và tài nguyên tính toán.

5. Thuật toán tối ưu Adam kết hợp những ưu điểm gì từ Momentum và RMSProp?
Adam kết hợp việc lưu trữ trung bình mượt mà của gradient bậc nhất từ quá khứ ($m_t$ như trong Momentum) và trung bình mượt mà của gradient bậc hai ($v_t$ như trong RMSProp), đồng thời áp dụng kỹ thuật hiệu chỉnh sai lệch thiên vị ($\hat{m}_t, \hat{v}_t$) giúp thuật toán thích ứng tốt với gradient thưa và bề mặt hàm mất mát không ổn định.


Kết luận

Báo cáo đồ án cuối kỳ "So sánh hiệu quả của các thuật toán tối ưu SGD, Momentum, RMSProp, và Adam trong việc huấn luyện mô hình mạng nơ-ron cho bài toán phân loại ảnh MNIST, sử dụng kỹ thuật L2 Regularization và Dropout" đã hệ thống hóa cơ sở lý thuyết về mạng nơ-ron và các giải thuật tối ưu hóa trong học sâu. Nhóm tác giả đã thiết kế và triển khai quy trình thực nghiệm hoàn chỉnh trên môi trường Python và Keras, áp dụng hiệu quả các kỹ thuật điều chuẩn nhằm đánh giá độ chính xác, giá trị hàm mất mát và thời gian huấn luyện. Công trình cung cấp tài liệu tham khảo có cấu trúc rõ ràng, phục vụ hữu ích cho việc nghiên cứu và ứng dụng các thuật toán tối ưu hóa trong xử lý dữ liệu hình ảnh.