DG-KD: Domain-Group KAN Detector
- The paper introduces DG-KD, a neural module that uses groupwise RBF activations and data-free replay to address catastrophic forgetting in face forgery detection.
- DG-KD partitions high-dimensional features into groups, applying domain-specific local activations to enable continual adaptation with minimal interference.
- Empirical results demonstrate high accuracy and minimal forgetting, achieving up to 99.48% memory reduction compared to traditional replay-based methods.
The Domain-Group KAN Detector (DG-KD) is a neural module designed for continual face forgery detection problems, enabling robust adaptation to new domains or forgery types while minimizing catastrophic forgetting. It is a component of the KAN-CFD framework, which integrates DG-KD with data-free replay and feature separation mechanisms to achieve high continual-learning performance with minimal storage and computational overhead (Zhang et al., 5 Aug 2025).
1. Rationale and Problem Setting
Continual face forgery detection is formulated as a domain-incremental learning scenario where the binary label space is constant, but the data distribution varies for each task/domain . The principal challenge is to learn new forgery types without incurring severe performance drops on previously seen domains—a phenomenon termed catastrophic forgetting.
DG-KD builds on Kolmogorov-Arnold Networks (KANs), which feature locally plastic activation functions: parameter updates in one input region only affect function behavior nearby, not globally. However, standard KANs employ B-spline activations, which are inefficient for high-dimensional image features and do not scale well. Conversely, conventional activations suitable for images (e.g., rational or nonlocal bases) lack the local adaptivity needed for continual learning.
DG-KD addresses this by combining locally supported radial basis function (RBF) activations with a group-wise domain partitioning strategy, facilitating high-dimensional feature modeling while retaining the locality essential for catastrophic forgetting mitigation.
2. Mathematical Structure
2.1 KAN Activation and Layer
A standard KAN layer maps an input to an output via a sum of learnable univariate functions : Each is parameterized by spline or similar local bases.
2.2 Local RBF Activation
DG-KD replaces spline bases with radial basis functions: where and denote center and width, enabling the activation to be significant only locally in feature space.
Correspondingly, a layer’s output can be written: 0
2.3 Domain Grouping Mechanism
The input feature vector is partitioned into 1 groups of size 2. Group index assignment is 3.
Within each group, all feature dimensions share a domain-specific local activation: 4 A DG-Layer for domain 5 applies groupwise RBFs, accumulated as: 6 with 7 a learnable weight matrix.
The overall DG-KD is the sum over DG-Layers: 8 This setup ensures that each task/domain utilizes a distinct, locally supported region of the activation manifold.
2.4 Locality and Knowledge Retention
Learning for a new domain only updates basis functions within the region associated with that domain: 9 This formalizes retention: earlier tasks’ knowledge, represented by nonoverlapping local regions, remains stable throughout subsequent task learning.
3. Integration with Model Pipeline
DG-KD is implemented as the head of a ConvNeXt-B convolutional neural network. The system operates as follows:
- ConvNeXt-B backbone extracts high-dimensional image features.
- These features are processed by the DG-KD module—a stack of DG-Layers (one per domain/task) with groupwise RBF activations.
- Each group shares local activation parameters within a DG-Layer.
- The classifier produces a binary real/fake output.
A canonical example with 0, 1, 2 demonstrates that each DG-Layer contains two RBF groups, each responsible for two input features, reducing redundancy in feature processing.
The feature memory is limited to 500 feature representatives, achieving data-free replay by storing latent features instead of raw input images, yielding a reported 3 memory reduction compared to conventional replay-based approaches (Zhang et al., 5 Aug 2025).
4. Objective Function and Optimization
Training comprises a composite loss: 4 where:
- 5 is binary cross-entropy,
- 6 for supervised contrastive loss 7,
- 8 for feature-level knowledge distillation loss 9,
- temperature for contrastive loss 0.
The supervised contrastive loss is: 1 where each real/fake per task is a unique class, giving 2 classes.
Knowledge distillation constrains backbone drift: 3
KAN Drift Compensation Projection (KDCP) is used for data-free feature alignment: 4
Optimization is performed using Adam (5, 6), with learning rates 7 for the main model and 8 for KDCP.
5. Locality, Plasticity, and Overlap Avoidance
Catastrophic forgetting in KANs arises when input features for multiple domains overlap, risking shared activation regions and global interference. DG-KD mitigates this in two ways:
- Local plasticity: The RBF basis ensures that updates for a given domain modify only its neighborhood in activation space:
9
with 0 for 1 if feature-space separation is achieved.
- Domain separation in feature space: FS-KDCP aligns memory features and supervised contrastive learning enforces separation, further minimizing domain overlap in DG-KD activations. This ensures that the network restructures only the relevant local regions during new-task learning, preserving stored knowledge from prior domains.
6. Empirical Findings and Ablations
DG-KD, within KAN-CFD, demonstrates strong resistance to forgetting and high accuracy on multiple continual learning benchmarks:
| Protocol | Metric | DG-KD/KAN-CFD Performance |
|---|---|---|
| Dataset-Incremental (2FF++, DFDC-P, DFD, CDF23) | Avg. Accuracy | 4 |
| Avg. Forgetting | 5 | |
| Forgery-Type Incremental (6Hybrid, FR, FS, EFS7) | Avg. AUC | 8 |
| Final Forgetting | 9 | |
| Long-Sequence (DF40, 10 tasks) | Highest accuracy, lowest forgetting | Yes |
Ablation studies reveal:
- Relying solely on 0 causes severe forgetting (1 by end of training).
- Adding 2 or 3 improves retention, with the full loss achieving the best results.
- DG-KD outperforms MLP, standard KAN, and GroupKAN, with its groupwise RBF architecture being critical for high-dimensional images and locality.
7. Technical Significance
DG-KD is a locality-preserving adaptation of Kolmogorov-Arnold Networks tailored for domain-incremental settings in high-dimensional image analysis. Its core technical contributions are:
- Replacement of B-splines with local RBFs to scale KANs to image features while preserving plasticity.
- Groupwise parameter sharing—partitioning features into groups each handled by a separate local basis—yielding parameter efficiency and locality.
- Integration with KDCP for data-free feature replay and enforced domain separation, maintaining high performance with minimal memory.
DG-KD exemplifies an architecture that combines local compositionality, continual-learning robustness, and computational feasibility for modern face forgery detection tasks, achieving strong empirical performance without the need for full data replay (Zhang et al., 5 Aug 2025).