Learning without Forgetting

Tóm tắt một câu

Learning without Forgetting (LwF) là phương pháp học liên tục (continual learning) đột phá sử dụng phân phối dự đoán của mô hình cũ trên dữ liệu tác vụ mới làm mục tiêu chưng cất tri thức (knowledge distillation loss), giúp mạng nơ-ron học thêm các khả năng mới mà không bị quên các tác vụ cũ (catastrophic forgetting) và hoàn toàn không cần lưu trữ hay truy cập dữ liệu quá khứ. PDF, tr. 1

Nguồn

Vấn đề paper giải quyết

Khi mở rộng hệ thống thị giác máy tính hoặc mô hình học sâu để thực hiện thêm các tác vụ mới theo thời gian, giả định thông thường là dữ liệu huấn luyện của toàn bộ các tác vụ cũ luôn có sẵn để huấn luyện đồng thời (Joint Training). Tuy nhiên, trong thực tế: PDF, tr. 1

  1. Dữ liệu cũ có thể quá đồ sộ, tốn kém chi phí lưu trữ và thời gian huấn luyện lại từ đầu.
  2. Dữ liệu cũ có thể bị ràng buộc bởi quyền riêng tư, bí mật thương mại, hoặc không được phép lưu trữ vĩnh viễn (proprietary/unrecorded data).
  3. Nếu chỉ đơn thuần tinh chỉnh (Fine-tuning) mạng nơ-ron trên dữ liệu mới, mô hình sẽ gặp hiện tượng Quên thảm khốc (Catastrophic Forgetting) — hiệu năng trên các tác vụ ban đầu sụt giảm nghiêm trọng.
  4. Nếu cố định đặc trưng (Feature Extraction - chỉ huấn luyện head mới), mô hình bảo toàn được tác vụ cũ nhưng hiệu năng trên tác vụ mới rất hạn chế vì các tầng chia sẻ không được tối ưu hóa cho tác vụ mới.

Paper đặt ra bài toán: Làm thế nào để học thêm tác vụ mới trên mạng CNN chia sẻ tham số, tối ưu hóa biểu diễn cho tác vụ mới, mà vẫn bảo toàn hiệu năng trên các tác vụ cũ khi KHÔNG CÓ bất kỳ dữ liệu huấn luyện nào của tác vụ cũ? PDF, tr. 1 PDF, tr. 2

Gap và đóng góp

  • Phương pháp LwF (Learning without Forgetting): Lần đầu tiên kết hợp ý tưởng Chưng cất tri thức (Knowledge Distillation - Hinton et al., 2015) vào bài toán Học liên tục mà không cần lưu trữ dữ liệu cũ (rehearsal-free continual learning).
  • Cơ chế ghi nhận phản hồi (Recorded Responses): Trước khi cập nhật trọng số, LwF dùng mạng nơ-ron ban đầu chạy feed-forward trên tập dữ liệu của tác vụ mới để thu được phân phối xác suất dự đoán trên các nhãn cũ. đóng vai trò là “nhãn mềm thay thế” (surrogate targets) để ràng buộc hành vi của mô hình. PDF, tr. 4 PDF, tr. 5
  • Quy trình huấn luyện hai giai đoạn (Warm-up & Joint-Optimize):
    • Warm-up: Đóng băng các tầng chia sẻ và tầng cũ , chỉ huấn luyện tầng tác vụ mới đến khi hội tụ để tránh việc khởi tạo ngẫu nhiên của tạo gradient lớn làm phá hỏng biểu diễn chia sẻ.
    • Joint-Optimize: Huấn luyện đồng thời tất cả các tham số với hàm mất mát kết hợp giữa hard-task loss (cho tác vụ mới) và distillation loss (cho tác vụ cũ). PDF, tr. 4
  • Hiệu quả thực nghiệm toàn diện:
    • Trên tác vụ mới, LwF vượt trội hơn Feature Extraction và thậm chí vượt cả Fine-tuning tiêu chuẩn (do hàm distillation loss đóng vai trò như một bộ điều hòa regularizer ngăn overfitting).
    • Trên tác vụ cũ, LwF giữ vững độ chính xác vượt bậc so với Fine-tuning, đạt kết quả tiệm cận với Joint Training (vốn được xem là cận trên lý thuyết sử dụng toàn bộ dữ liệu cũ). PDF, tr. 6 PDF, tr. 7
  • Phân tích điều kiện suy thoái (Failure Modes): Chỉ ra giới hạn bản chất của LwF khi phân phối dữ liệu mới quá khác biệt so với dữ liệu cũ (ví dụ: ImageNet MNIST hoặc Places CUB), dẫn đến việc dữ liệu mới không kích hoạt được các đặc trưng quan trọng của tác vụ cũ. PDF, tr. 7

Bài toán/formalization

Mạng nơ-ron tích chập (CNN) ban đầu bao gồm:

  • : Các tham số chia sẻ (ví dụ: 5 tầng tích chập và 2 tầng fully-connected trong AlexNet).
  • : Các tham số chuyên biệt cho tác vụ cũ (output layer và trọng số kết nối với tầng chia sẻ cuối cùng).
  • Mục tiêu: Thêm bộ tham số chuyên biệt cho tác vụ mới và học trên tập dữ liệu mới sao cho mô hình hoạt động tốt trên cả tác vụ cũ lẫn tác vụ mới, mà không cần dữ liệu cũ . PDF, tr. 4

Quy trình toán học

  1. Khởi tạo và ghi nhận phản hồi cũ: Khởi tạo ngẫu nhiên .

  2. Định nghĩa đầu ra dự đoán của mô hình hiện tại:

    • Dự đoán trên tác vụ cũ:
    • Dự đoán trên tác vụ mới:
  3. Hàm mục tiêu tổng quát: Trong đó:

    • là hàm điều hòa suy giảm trọng số (weight decay, mặc định 0.0005).
    • là trọng số cân bằng giữa tác vụ cũ và mới (mặc định ). PDF, tr. 4 PDF, tr. 5

Các thành phần hàm mất mát

  1. Mất mát tác vụ mới (): Sử dụng cross-entropy đa lớp thông thường: với và là one-hot ground-truth vector. PDF, tr. 4

  2. Mất mát chưng cất tác vụ cũ (): Sử dụng modified cross-entropy với nhiệt độ làm mềm : Trong đó xác suất được làm mềm ở nhiệt độ (paper chọn thông qua grid search): Việc chọn giúp khuếch đại các xác suất nhỏ, ép mô hình mới phải ghi nhớ sự tương đồng giữa các lớp cũ thay vì chỉ chú ý vào lớp có xác suất cao nhất. PDF, tr. 4 PDF, tr. 5

Phương pháp

[Dữ liệu tác vụ mới X_n]
        │
        ├─────────────────────────────────────────┐
        ▼ (Forward qua mạng cũ θ_s, θ_o)          ▼
[Ghi nhận nhãn mềm cũ Y_o]                 [Nhãn thật tác vụ mới Y_n]
        │                                         │
        ▼                                         ▼
[Forward qua mạng đang học]               [Forward qua mạng đang học]
  sinh ra logits cũ z_o                     sinh ra logits mới z_n
        │                                         │
        ▼ (Softmax tại T=2)                       ▼ (Softmax tại T=1)
  xác suất mềm ŷ'_o                         xác suất dự đoán ŷ_n
        │                                         │
        ▼                                         ▼
[Distillation Loss L_old]                  [Task Loss L_new]
        │                                         │
        └────────────────────┬────────────────────┘
                             ▼
               Loss tổng = L_new + λ_o L_old
                             │
                             ▼
         [Cập nhật đồng thời θ_s, θ_o, θ_n]

So sánh vị thế với các hướng tiếp cận khác (Figure 1, Table 1)

Phương phápDữ liệu tác vụ cũHiệu năng tác vụ mớiHiệu năng tác vụ cũBộ nhớ lưu trữThời gian huấn luyệnThời gian suy luận
Fine-TuningKhôngTốtKém (Quên nặng)Vừa phảiNhanhNhanh
Duplicating & Fine-TuningKhôngTốtHoàn hảoLớn ( networks)NhanhChậm ( lần)
Feature ExtractionKhôngTrung bìnhHoàn hảoVừa phảiRất nhanhNhanh
Joint TrainingCó (Bắt buộc)Tốt nhấtRất tốtRất lớn (Lưu data cũ)Rất chậmNhanh
LwF (Đề xuất)KhôngTốt nhấtRất tốtVừa phảiNhanhNhanh

PDF, tr. 2

Protocol fingerprint

TrườngGiá trị trong LwF
Tác vụ thực nghiệmPhân loại ảnh (Image Classification) & Theo dõi đối tượng video (Tracking)
Bộ dữ liệu tác vụ cũImageNet (1.000 lớp, ~1.2M ảnh) hoặc Places365-standard (365 lớp, ~1.8M ảnh)
Bộ dữ liệu tác vụ mớiPASCAL VOC 2012 (20 lớp, 5.717 ảnh train), CUB-200-2011 (200 loài chim, 5.994 ảnh train), MIT Indoor Scenes (67 cảnh, 5.360 ảnh train), MNIST (10 số viết tay)
Kiến trúc mạngAlexNet (5 conv + 3 fc) làm mạng chính; mở rộng kiểm chứng trên VGG-16
Tham số chưng cấtNhiệt độ ; trọng số cân bằng
Tối ưu hóaSGD, momentum 0.9, weight decay 0.0005, dropout trên fully-connected layers
Giai đoạn huấn luyệnBước 1: Warm-up chỉ train (lr = 0.001); Bước 2: Joint-optimize train (lr = 0.0001, giảm 10 lần sau mỗi số epochs cố định)
Độ đoMean Average Precision (mAP) cho VOC; Top-1 Accuracy cho tất cả các tập dữ liệu còn lại
Phân loại kết quảBáo cáo từ bài báo gốc (reported/observed). Không tái lập nội bộ

Kết quả chính

1. Hiệu năng trong kịch bản thêm một tác vụ đơn lẻ (Bảng 1)

Bảng 1(a) báo cáo độ chênh lệch hiệu năng so với LwF trên kiến trúc AlexNet. Giá trị âm nghĩa là phương pháp so sánh kém hơn LwF: PDF, tr. 7

Phương phápImageNet VOCImageNet CUBImageNet ScenesPlaces365 VOCPlaces365 CUBPlaces365 ScenesImageNet MNIST
Cũ / MớiCũ / MớiCũ / MớiCũ / MớiCũ / MớiCũ / MớiCũ / Mới
LwF (Bản thân)56,2 / 76,154,7 / 57,755,9 / 64,550,6 / 70,247,9 / 34,850,9 / 75,249,8 / 99,3
Fine-Tuning-0,9 / -0,3-3,8 / -0,7-2,0 / -0,8-2,2 / +0,1-4,6 / +1,0-2,1 / -1,7-2,8 / 0,0
LFL (Jung et al.)0,0 / -0,4-1,9 / -2,6-0,3 / -0,9+0,2 / -0,7+0,7 / -1,7-0,2 / -0,5-2,9 / -0,6
Fine-tune FC+0,5 / -0,7+0,2 / -3,9+0,6 / -2,1+0,5 / -1,3+1,8 / -4,9+0,3 / -1,1+7,0 / -0,2
Feature Extraction+0,8 / -0,5+2,3 / -5,2+1,2 / -3,3+1,1 / -1,4+3,8 / -12,3+0,8 / -1,7+7,3 / -0,8
Joint Training+0,7 / -0,2+0,6 / -1,1+0,5 / -0,6+0,7 / 0,0+2,3 / +1,5+0,3 / -0,3+7,2 / 0,0

Quan sát mấu chốt:

  1. LwF vượt Fine-tuning trên tác vụ mới: Trên hầu hết các cặp tác vụ tương đồng (ImageNet VOC, ImageNet Scenes, Places Scenes), LwF đạt độ chính xác tác vụ mới cao hơn cả Fine-tuning ( đến ). Tín hiệu chưng cất từ tác vụ cũ đóng vai trò như một bộ điều hòa mạnh mẽ chống overfitting.
  2. Khả năng giữ tác vụ cũ: LwF giảm thiểu tối đa hiện tượng quên so với Fine-tuning (Fine-tuning làm sụt giảm từ 2% đến gần 5% độ chính xác tác vụ cũ).
  3. Tiệm cận Joint Training: LwF chỉ thua Joint Training khoảng trên tác vụ cũ, dù hoàn toàn không dùng một mẫu dữ liệu cũ nào.

2. Kịch bản thêm tuần tự nhiều tác vụ (Figure 4)

Khi tuần tự thêm 3 phần dữ liệu của VOC vào Places365 (Places ) hoặc Scenes vào ImageNet: PDF, tr. 8

  • Độ chính xác của Fine-tuning trên tác vụ gốc suy giảm dốc đứng qua từng tác vụ mới.
  • LwF duy trì đường suy thoái rất phẳng, bảo vệ tác vụ gốc tốt hơn hẳn và bám sát đường biểu diễn của Joint Training.

3. Tác dụng của bước Warm-up (Bảng 2b)

Thiết lậpImageNet CUB (Cũ / Mới)ImageNet Scenes (Cũ / Mới)Places365 VOC (Cũ / Mới)
LwF có Warm-up54,7 / 57,755,9 / 64,550,6 / 70,2
LwF không Warm-up53,5 / 59,955,2 / 64,950,4 / 70,0
Fine-tuning có Warm-up50,9 / 57,053,9 / 63,848,4 / 70,3
Fine-tuning không Warm-up42,5 / 59,8 (-8,4% cũ)49,8 / 63,9 (-4,1% cũ)42,3 / 70,0 (-6,1% cũ)

Kết luận: Nếu Fine-tuning không có bước Warm-up, tác vụ cũ bị xóa sổ nghiêm trọng (sụt giảm ). LwF bền vững hơn nhiều: ngay cả khi không có Warm-up, độ sụt giảm của tác vụ cũ chỉ là . PDF, tr. 9

Hạn chế, giả định, failure modes

  • Sự lệch pha phân phối (Distribution Mismatch / Dissimilar Tasks): LwF dựa trên giả định rằng việc ép phản hồi của mạng mới trên giống với mạng cũ sẽ gián tiếp bảo vệ vùng biểu diễn của tác vụ cũ. Nếu dữ liệu tác vụ mới hoàn toàn xa lạ so với (như ImageNet MNIST):
    • Ảnh số viết tay đen trắng trên nền đen hoàn toàn không kích hoạt các bộ lọc kết cấu, màu sắc, hình thái tự nhiên của ImageNet.
    • Do đó, distillation loss trên MNIST không cung cấp được bất kỳ ràng buộc có ý nghĩa nào cho các vùng không gian đặc trưng của ImageNet, khiến tác vụ cũ bị tụt 7,2% so với Joint Training. PDF, tr. 7
  • Tích lũy sai số khi chuỗi tác vụ quá dài (Error Accumulation): Vì teacher của task sau chính là student của task trước, các sai lệch nhỏ trong việc xấp xỉ phân phối sẽ bị khuếch đại dần qua nhiều task liên tiếp nếu không có dữ liệu gốc để hiệu chỉnh lại ranh giới quyết định.
  • Phụ thuộc vào kiến trúc chia sẻ toàn phần (Shared Backbone): LwF buộc toàn bộ các tác vụ phải chen chúc trong cùng một tập trọng số , dễ dẫn đến hiện tượng tắc nghẽn dung lượng (capacity bottleneck) khi số lượng tác vụ tăng lên hàng chục hoặc hàng trăm. PDF, tr. 11

Đánh giá từ evidence

  • Ý nghĩa lịch sử: LwF là bài báo kinh điển mở ra phân nhánh Distillation-based Continual Learning không cần bộ nhớ đệm (rehearsal-free). Nó chứng minh rằng không nhất thiết phải lưu dữ liệu cũ thì mới chống được quên.
  • Bằng chứng chặt chẽ: So sánh công bằng trên cả mạng AlexNet và VGG-16, phân tích kỹ lưỡng ảnh hưởng của Warm-up, kích thước tập dữ liệu (3% đến 100%), và trọng số .

Diễn giải học tập

1. Cơ chế cốt lõi: Vì sao Teacher chỉ biết nhãn cũ vẫn bảo vệ được mô hình?

Trong LwF, Teacher không cần biết gì về các nhãn mới. Khi một ảnh của task mới đi qua Teacher, Teacher sẽ dự đoán phân phối xác suất trên toàn bộ các nhãn cũ.

  • Phân phối này phản ánh vị trí tương đối của đối với các ranh giới quyết định cũ (decision boundaries).
  • Bằng cách ép Student phải tạo ra phân phối y hệt trên các nhãn cũ, Student bị cấm không được làm biến dạng các góc chiếu hoặc dịch chuyển các mặt phẳng phân chia của các lớp cũ trong không gian đặc trưng chung .

2. Mối liên hệ sống còn với Continual Relation Extraction (CRE / TAPTA)

Trong đề tài nghiên cứu Continual Relation Extraction with Task-Aware Prompt Adaptation (TAPTA), LwF là nguồn cảm hứng trực tiếp cho nhánh Teacher-Student Knowledge Distillation.

Trả lời trực tiếp câu hỏi thiết kế trong Reading Queue và Paper Decisions:

“Hiểu teacher cũ trong continual learning: Teacher dùng để giữ old-label behavior như thế nào? Và áp dụng vào TAPTA ra sao?”

Bài học từ LwF áp dụng vào TAPTA:

  1. Khắc phục điểm yếu Distribution Mismatch của LwF:
    • Trong CRE, mỗi task mang đến các quan hệ mới trong các câu văn bản khác nhau. Nếu chỉ dùng LwF thuần túy trên Backbone của ngôn ngữ (PLM), câu của task mới có thể không chứa ngữ cảnh kích hoạt các quan hệ cũ.
    • Do đó, trong TAPTA, ta không thể chỉ dựa vào LwF trên toàn bộ câu mới. Cần kết hợp:
      • Prompt Isolation: Dùng Router để kích hoạt prompt riêng cho từng task, giảm tải việc chia sẻ toàn bộ trọng số như CNN của LwF.
      • Similarity Distillation với Prototype cũ: Thay vì ép khớp logits của một linear head (như LwF), ta ép student giữ nguyên phân phối độ tương đồng giữa đặc trưng câu và các relation prototypes cũ.
  2. Kỹ thuật Warm-up:
    • Bài học từ Bảng 2(b) của LwF cho thấy: khi sang task mới, bắt buộc phải có giai đoạn Warm-up (chỉ train module mới / prompt mới của task hiện tại trước, đóng băng router và backbone cũ) trước khi mở khóa joint-optimization. Nếu không có warm-up, gradient khởi tạo ngẫu nhiên từ task mới sẽ phá hủy ngay lập tức các biểu diễn cũ đã ổn định.

Ghi chú cá nhân

(Bảo lưu ghi chú định hướng đề tài)

  • Student-Teacher KD trong đề tài là hậu duệ trực tiếp của LwF. Paper này giúp viết phần nền cho:
    • vì sao Teacher chỉ biết old labels vẫn có ích;
    • vì sao KD trên dữ liệu task mới có thể giữ old behavior;
    • giới hạn của KD khi dữ liệu task mới không kích hoạt old decision boundary.

Câu hỏi review

Câu 1: Ý tưởng cốt lõi của LwF là gì, và tại sao nó không cần dữ liệu của tác vụ cũ?

Gợi ý trả lời: LwF sử dụng chính dữ liệu của tác vụ mới , đưa qua mô hình cũ để ghi nhận phân phối dự đoán trên các nhãn cũ . Khi huấn luyện mô hình mới, hàm mất mát chưng cất tri thức (Knowledge Distillation loss tại ) ép mô hình mới phải tái hiện lại phản hồi này, qua đó duy trì hành vi trên các nhãn cũ mà không cần lưu trữ bất kỳ mẫu dữ liệu cũ nào.

Câu 2: Bước Warm-up trong LwF đóng vai trò gì, và điều gì xảy ra nếu bỏ qua bước này?

Gợi ý trả lời: Warm-up là bước đóng băng các tầng chia sẻ và tầng cũ , chỉ huấn luyện các tham số mới đến khi hội tụ. Nếu bỏ qua bước này, các trọng số ngẫu nhiên ban đầu của sẽ sinh ra gradient rất lớn làm xáo trộn các tầng chia sẻ , khiến tác vụ cũ bị quên nghiêm trọng (trong thực nghiệm Fine-tuning, bỏ warm-up làm sụt giảm độ chính xác tác vụ cũ).

Câu 3: Khi nào LwF gặp thất bại nghiêm trọng nhất (Failure Mode)?

Gợi ý trả lời: LwF thất bại khi phân phối dữ liệu của tác vụ mới hoàn toàn xa lạ và không có điểm chung với dữ liệu cũ (ví dụ: chuyển từ ImageNet sang ảnh chữ số viết tay MNIST). Khi đó, các mẫu mới không kích hoạt được các đặc trưng ngữ nghĩa của tác vụ cũ, khiến tín hiệu distillation trên không thể bảo vệ được các ranh giới phân loại của tác vụ cũ.

Câu 4: Tại sao trong nhiều kịch bản, LwF lại đạt kết quả trên tác vụ mới cao hơn cả Fine-tuning thông thường?

Gợi ý trả lời: Bởi vì hàm distillation loss trên các nhãn cũ đóng vai trò như một bộ điều hòa (regularizer) mạnh mẽ. Nó ngăn không cho mạng nơ-ron bị overfit vào tập dữ liệu hữu hạn của tác vụ mới, giữ cho không gian đặc trưng tổng quát hơn.

Evidence map

Luận điểm / Nội dungTrang PDFBằng chứng cụ thể trong nguồn
Đặt bài toán & Bảng so sánh 5 phương phápPDF, tr. 2Figure 1: So sánh Fine-tuning, Duplicating, Feature Extraction, Joint Training và LwF
Kiến trúc mạng chia sẻ & Tách biệt PDF, tr. 3Figure 2: Sơ đồ kiến trúc các tầng chia sẻ và tầng tác vụ
Thuật toán LwF & Công thức Distillation Loss ()PDF, tr. 4-5Mục 3, Eq. 1-4 & Figure 3: Khởi tạo, Warm-up và Joint optimization
Thiết lập thực nghiệm & Bộ dữ liệu (AlexNet, VGG)PDF, tr. 6Mục 4: ImageNet, Places365, VOC, CUB, Scenes, MNIST
Bảng kết quả chính trên đơn tác vụ (Bảng 1)PDF, tr. 7Table 1: LwF vượt Fine-tuning trên task mới và giữ vững task cũ
Phân tích thất bại trên tác vụ lệch pha (MNIST)PDF, tr. 7Mục 4.1: ImageNet MNIST tụt 7,2% do thiếu kích hoạt đặc trưng cũ
Kết quả thêm tuần tự nhiều tác vụPDF, tr. 8Figure 4: Đường cong suy thoái hiệu năng qua chuỗi 3 task liên tiếp
Ảnh hưởng của bước Warm-upPDF, tr. 9Table 2(b): Fine-tuning mất nếu không warm-up; LwF duy trì ổn định
Đánh đổi hiệu năng qua trọng số PDF, tr. 10Figure 7: Đồ thị Pareto trade-off giữa tác vụ cũ và tác vụ mới

Liên kết