Detecting Diffusion-generated Images via Dynamic Assembly ForestsDetecting Diffusion-generated Images via Dynamic Assembly Forests
Published on arXiv
2604.09106
Output Integrity Attack
OWASP ML Top 10 — ML09
Key Finding
Achieves competitive detection performance with significantly fewer parameters and lower computational cost than DNN-based methods, enabling GPU-free deployment
Dynamic Assembly Forest (DAF)
Novel technique introduced
Diffusion models are known for generating high-quality images, causing serious security concerns. To combat this, most efforts rely on deep neural networks (e.g., CNNs and Transformers), while largely overlooking the potential of traditional machine learning models. In this paper, we freshly investigate such alternatives and proposes a novel Dynamic Assembly Forest model (DAF) to detect diffusion-generated images. Built upon the deep forest paradigm, DAF addresses the inherent limitations in feature learning and scalable training, making it an effective diffusion-generated image detector. Compared to existing DNN-based methods, DAF has significantly fewer parameters, much lower computational cost, and can be deployed without GPUs, while achieving competitive performance under standard evaluation protocols. These results highlight the strong potential of the proposed method as a practical substitute for heavyweight DNN models in resource-constrained scenarios. Our code and models are available at https://github.com/OUC-VAS/DAF.
Key Contributions
- Novel Dynamic Assembly Forest (DAF) detector for diffusion-generated images using traditional ML instead of DNNs
- Significantly reduced computational cost and parameter count compared to CNN/Transformer baselines while maintaining competitive detection accuracy
- GPU-free deployment suitable for resource-constrained scenarios
🛡️ Threat Analysis
Detects AI-generated images (specifically diffusion model outputs) to verify content authenticity and provenance — core ML09 output integrity task.