Radar Live
✦ Ask AI
AI INTELLIGENCE & SIGNALS

A Spectral Theory of Grokking: Weight Decay induces Feature Learning

Phân tích đã xử lý trước và các nguồn liên quan.

Tín hiệu Radar

65 RADAR
Research Machine Learning Theory · 1 nguồn đối chiếu · 2026-09-22 16:38:16 +0000 UTC

A Spectral Theory of Grokking: Weight Decay induces Feature Learning

Nguồn: arXiv AI
↗

Nghiên cứu thiết lập một lý thuyết phổ định lượng giải thích hiện tượng grokking (khái quát hóa muộn sau khi đã ghi nhớ dữ liệu huấn luyện) thông qua sự cạnh tranh giữa phần dư sau ghi nhớ và L2 weight decay. Trong các mạng thuần nhất dùng squared loss, phần dư kích hoạt sự phát triển của Neural Tangent Kernel (NTK) theo các hướng phổ liên quan đến tác vụ, thúc đẩy quá trình chuyển từ lazy sang rich learning. Nghiên cứu chứng minh thang thời gian grokking tỷ lệ nghịch với tích của learning rate và weight decay, đồng thời cảnh báo ngưỡng decay tới hạn có thể làm chậm hoặc ngăn chặn việc học đặc trưng. Các kết quả được kiểm chứng bằng thực nghiệm trên MLP và One-block Transformer ở bài toán modular addition.

✦ TÁC ĐỘNG & GIÁ TRỊ THỰC TIỄN

Nghiên cứu cung cấp cơ sở toán học cơ chế cho hiện tượng grokking vốn được coi là bí ẩn, giúp các kỹ sư và nhà nghiên cứu AI hiểu rõ và dự đoán được mối quan hệ giữa learning rate và weight decay để kiểm soát quá trình tổng quát hóa, tránh lãng phí compute khi huấn luyện mô hình suy luận.

Developer 62
Business 50
Novelty 85
Actionable 62

Nội dung thu thập đã chuẩn hóa

Collected Evidence

Nội dung văn bản được dùng làm dữ liệu đối chứng cho mô hình AI, không phải chỉ thị hệ thống.

In grokking an early fit to the training data separates from a much later improvement in generalization. During this delay, training can move from a fixed neural tangent kernel (NTK) regime to one in which task-relevant kernel eigendirections continue to evolve. We provide a quantitative theory for how this transition from lazy to rich learning can produce delayed generalization. For homogeneous networks trained with squared loss and $L_2$ weight decay, we show that a finite residual remains after memorization, with larger residual fractions in target components associated with smaller NTK eigenvalues. These residuals feed back into the dynamics of the NTK itself, and projecting the resulting dynamics onto task-relevant spectral directions yields a reduced system in which residual-driven kernel growth competes with weight decay. This system predicts that the grokking timescale is controlled by the product of learning rate and weight decay, that feature learning slows logarithmically near a critical decay above which task-aligned NTK structure can no longer support generalization, and that stronger decay can prevent fitting altogether. We test these predictions in modular addition. In a homogeneous MLP, task-aligned Fourier structure continues to emerge in the NTK after training accuracy has saturated, and an 84$\times$90-grid of trained networks across varying learning rate and weight decay recovers the predicted phase geometry and inverse-product scaling of the generalization time with learning rate and weight decay. A one-block Transformer shows similar macroscopic phase structure in a 42$\times$45-grid, as well as the same transition-time scaling despite violating exact homogeneity. Together, these results provide a mechanistic derivation connecting post-fit feature learning to both the onset of generalization and its phase structure in the learning rate and weight decay plane.

Các nguồn đối chiếu cho sự kiện này

1 nguồn
Thông tin phân tích AI & Model Details ▾
Provider: openai-compatible · Model: gemini-3.8-flash-high · Version: analysis-v1 · Time: 2026-09-23 09:55:18 +0000 UTC
{"tags": ["grokking", "neural-tangent-kernel", "weight-decay", "feature-learning", "deep-learning-theory", "generalization", "transformers"], "risks": ["Mô hình lý thuyết dựa trên giả định mạng thuần nhất và hàm mất mát squared loss; việc áp dụng trực tiếp cho mạng không thuần nhất và cross-entropy cần thêm nghiên cứu hiệu chỉnh.", "Thiết lập weight decay quá cao có thể vượt qua điểm tới hạn, làm triệt tiêu hoàn toàn khả năng fit dữ liệu của mô hình."], "category": "Research", "entities": ["Neural Tangent Kernel (NTK)", "Transformer", "MLP", "Fourier Transform"], "summary_vi": "Nghiên cứu thiết lập một lý thuyết phổ định lượng giải thích hiện tượng grokking (khái quát hóa muộn sau khi đã ghi nhớ dữ liệu huấn luyện) thông qua sự cạnh tranh giữa phần dư sau ghi nhớ và L2 weight decay. Trong các mạng thuần nhất dùng squared loss, phần dư kích hoạt sự phát triển của Neural Tangent Kernel (NTK) theo các hướng phổ liên quan đến tác vụ, thúc đẩy quá trình chuyển từ lazy sang rich learning. Nghiên cứu chứng minh thang thời gian grokking tỷ lệ nghịch với tích của learning rate và weight decay, đồng thời cảnh báo ngưỡng decay tới hạn có thể làm chậm hoặc ngăn chặn việc học đặc trưng. Các kết quả được kiểm chứng bằng thực nghiệm trên MLP và One-block Transformer ở bài toán modular addition.", "key_changes": ["Đề xuất lý thuyết phổ định lượng giải thích cơ chế chuyển dịch từ lazy learning (NTK cố định) sang rich learning (học đặc trưng) trong hiện tượng grokking.", "Chứng minh phần dư sau ghi nhớ (finite residual) tương tác ngược lại động lực học của NTK và cạnh tranh trực tiếp với L2 weight decay.", "Xác lập quy luật tỷ lệ: thang thời gian xuất hiện grokking tỷ lệ nghịch với tích của learning rate và weight decay (inverse-product scaling).", "Phát hiện ngưỡng suy giảm trọng số tới hạn (critical decay), nếu vượt quá sẽ ngăn cản khả năng học đặc trưng hoặc khiến mô hình không thể khớp dữ liệu huấn luyện.", "Kiểm chứng thực nghiệm thành công trên bài toán cộng modulo với mạng MLP thuần nhất (lưới 84x90) và mô hình Transformer 1-block (lưới 42x45)."], "sub_category": "Machine Learning Theory", "novelty_score": 85, "business_score": 42, "research_score": 92, "why_it_matters": "Nghiên cứu cung cấp cơ sở toán học cơ chế cho hiện tượng grokking vốn được coi là bí ẩn, giúp các kỹ sư và nhà nghiên cứu AI hiểu rõ và dự đoán được mối quan hệ giữa learning rate và weight decay để kiểm soát quá trình tổng quát hóa, tránh lãng phí compute khi huấn luyện mô hình suy luận.", "developer_score": 60, "importance_score": 80, "possible_use_cases": ["Tối ưu hóa đồng thời learning rate và weight decay để rút ngắn thời gian huấn luyện kích hoạt khả năng tổng quát hóa trong các tác vụ suy luận/toán học.", "Dự đoán và kiểm soát pha chuyển tiếp tổng quát hóa của Transformer thay vì dựa hoàn toàn vào thử nghiệm mò mẫm tốn kém compute.", "Phát triển các công cụ phân tích phổ NTK nhằm theo dõi sức khỏe và tiến trình học biểu diễn của mô hình nơ-ron."], "actionability_score": 62}
← Trang trước Trang 1