DiffusionFake: Enhancing Generalization in Deepfake Detection via Guided Stable Diffusion

The rapid progress of Deepfake technology has made face swapping highly realistic, raising concerns about the malicious use of fabricated facial content. Existing methods often struggle to generalize to unseen domains due to the diverse nature of facial manipulations. In this paper, we revisit the generation process and identify a universal principle: Deepfake images inherently contain information from both source and target identities, while genuine faces maintain a consistent identity. Building upon this insight, we introduce DiffusionFake, a novel plug-and-play framework that reverses the generative process of face forgeries to enhance the generalization of detection models. DiffusionFake achieves this by injecting the features extracted by the detection model into a frozen pre-trained Stable Diffusion model, compelling it to reconstruct the corresponding target and source images. This guided reconstruction process constrains the detection network to capture the source and target related features to facilitate the reconstruction, thereby learning rich and disentangled representations that are more resilient to unseen forgeries. Extensive experiments demonstrate that DiffusionFake significantly improves cross-domain generalization of various detector architectures without introducing additional parameters during inference. Our Codes are available in https://github.com/skJack/DiffusionFake.git.

Paper

Similar papers

Peer review

Reviewer b2gp7/10 · confidence 5/52024-07-06

Summary

This paper introduces DiffusionFake to address the challenge of generalization in face forgery detection by revisiting the generative process of deepfakes. DiffusionFake reverses the generative process by injecting features into a pre-trained Stable Diffusion model to reconstruct source and target images. The plug-and-play framework integrates with existing detectors, improving cross-domain generalization without adding inference parameters. Experiments demonstrate significant performance improvements across various architectures.

Strengths

1. This paper presents a novel approach by designing a method from the deepfake creation perspective, introducing a reverse method, and utilizing pre-trained Stable Diffusion knowledge to address information loss during reconstruction. It effectively extracts features using a guide module to obtain source-related and target-related information through reconstruction, which is logically sound. 2. The paper proposes a plug-and-play functionality that enhances the generalization ability of multiple detection networks without adding parameters, facilitating deployment. 3. This paper is well-written. The authors offer an easy-to-follow presentation with a well-structured format, while important visualizations and figures are clearly provided.

Weaknesses

1. The training set seemingly contains instances where the source and target ground truth are identical for the same image. The paper should address how this trade-off is handled. 2. The discussion on the weight module could be clearer and more detailed. 3. In the visualization figures (Fig. 3), the reconstructed source images appear to differ from the ground truth (even the training sample). The authors should explain the reason for this discrepancy. 4. More ablation studies should be conducted. For example, the structure of the feature-transform module is insufficiently evaluated.

Questions

1. Would the performance improve and convergence speed up if the parameters of Stable Diffusion were unfrozen during training? 2. How significant is the impact of Stable Diffusion's pre-trained knowledge on the overall method? Have other pre-trained diffusion methods been tested? For more questions please refer to the Weaknesses.

Rating

7

Confidence

5

Soundness

3

Presentation

4

Contribution

3

Limitations

The authors have adequately addressed the limitations.

Authorsrebuttal2024-08-10

Dear Reviewer b2gp, We greatly appreciate your valuable feedback and the time you've taken to review our manuscript. We are sincerely grateful for your positive evaluation of our paper. Additionally, we would like to express our appreciation for the constructive experimental suggestions you've provided, which will undoubtedly make our paper more comprehensive. As we approach the end of the discussion period, we eagerly await your thoughts on our response. We sincerely hope that our revisions align with your expectations. If there are any remaining concerns or aspects that require clarification, we are ready to address them as soon as possible. Best regards, The Authors

Reviewer b2gp2024-08-10

Thanks for the response. The authors have adequately addressed my concern. I wish the analysis and experiments in the rebuttal could be appropriately incorporated into the manuscript to make it more comprehensive and compact. I look forward to seeing the updated results and analysis in the camera-ready version.

Authorsrebuttal2024-08-14

We appreciate your recognition of our work and rebuttal. We will further improve the paper based on your suggestions. Thank you for your efforts in reviewing our paper.

Reviewer NDSD5/10 · confidence 5/52024-07-06

Summary

This paper proposes DiffusionFake which can harnesses the power of Stable Diffusion to guide the forgery detector in learning disentangled source and target features inherent in Deepfakes. The features of the detection networks are processed through the target and source transformation modules, and then injected into the Stable Diffusion to reconstruct the source and target images. Through the proposed strategy, DiffusionFake can enhance its ability to handle unseen forgeries without compromising efficiency.

Strengths

The idea of using SD-based reconstruction to enhance detection accuracy is rational.

Weaknesses

1. The weighting module is supervised by the similarity between z and $z_s$/$z_t$. However, the latent representation of stable diffusion is not linear. I think the assumption of this constraint is not always correct. 2. There is no analysis for the influence of the loss weight in Eq. 13 3. There is no cross-model evaluation, i.e., the samples during testing are generated by a method that has not been seen during the training. 4. The evaluation is only conducted on the images with low resolution.

Questions

1. How to decide the optimal loss weights in Eq. 13? 2. How is the performance if we conduct cross-model evaluation? 3. How is the performance when the input image has high resolution?

Rating

5

Confidence

5

Soundness

2

Presentation

2

Contribution

2

Limitations

If some attackers have known the training details of the detector and use it as the metric to enhance the performance of deepfake, it will be harder to detect them.

Authorsrebuttal2024-08-10

Dear Reviewer NDSD, Thank you for your invaluable efforts and constructive feedback on our manuscript. We greatly appreciate your positive evaluation of our paper's rationale. We have endeavored to provide comprehensive responses to the concerns you raised in your review. As the discussion period draws to a close, we eagerly await your thoughts on our response. We sincerely hope that our revisions align with your expectations. If there are any remaining concerns or aspects that require clarification, we are ready to address them as soon as possible. Best regards, The Authors

Reviewer NDSD2024-08-13

Thanks for your response

Some of my concerns have been resolved, and I decide to raise my score.

Authorsrebuttal2024-08-14

We sincerely appreciate the reviewer's recognition of our work. We are grateful for your careful consideration of our rebuttal and the time you've invested in evaluating our research.

Reviewer VbH67/10 · confidence 5/52024-07-13

Summary

This paper adopts a novel plug-and-play framework that reverses the generative process of face forgeries to enhance the generalization of detection models. Extensive experimental results from several datasets demonstrate that this method has achieved very competitive performance.

Strengths

- The motivation of this paper is clear and the approach to implementing it is straightforward and understandable. - This paper attempts to introduce Stable Diffusion into deepfake detection, unifying the generation model and the detection model end-to-end, which is very interesting and innovative, and provides a new perspective for this field. - The overall structure of this paper is relatively reasonable and the context of the paper is relatively clear.

Weaknesses

- Some recently published papers about enhancing generalization in deepfake detection are not cited and discussed, such as: [1] Dong, Shichao, et al. "Implicit identity leakage: The stumbling block to improving deepfake detection generalization." Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2023. [2] Yu, Bingyao, et al. "Discrepancy-aware meta-learning for zero-shot face manipulation detection." IEEE Transactions on Image Processing (2023). [3] Chen, Liang, et al. "Self-supervised learning of adversarial example: Towards good generalizations for deepfake detection." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2022. - Many of the experimental results in Table 1 provided by the authors in the paper are inconsistent with or may be lower than the experimental results in the original paper. It is best for the authors to provide detailed, clear and convincing explanations. - In order to better understand the performance of the proposed method, it would be best if the authors could provide sample visualization results of misprediction and give corresponding theoretical analysis. - Please unify the format of references. At least make sure that the reference formats of conferences and journals are consistent. - If the author can answer and revise the relevant questions in the final version, I will consider increasing the final score in the next round.

Questions

Please standardize the capitalization of English letters in the references. Many abbreviations of proper nouns are incorrect, such as Aunet (AUNet).

Rating

7

Confidence

5

Soundness

3

Presentation

3

Contribution

3

Limitations

The authors have adequately addressed the limitations and potential negative societal impact of their work.

Authorsrebuttal2024-08-10

Dear Reviewer VbH6, We are deeply grateful for your thorough review and insightful comments on our paper. We greatly appreciate your evaluation of our work and assure you that we will comprehensively incorporate your feedback into our revised paper, enhancing its content and quality. As the discussion period draws to a close, we eagerly await your thoughts on our response. We sincerely hope that our revisions align with your expectations. If there are any remaining concerns or aspects that require clarification, we are ready to address them promptly. Best regards, The Authors

Authorsrebuttal2024-08-14

We sincerely appreciate the reviewer's recognition of our work. Your positive feedback is greatly encouraging. We are grateful for your careful consideration of our rebuttal and the time you've invested in evaluating our research.

Reviewer tkET7/10 · confidence 3/52024-07-25

Summary

The paper investigates the task of deepfake (especially, face swapping) identification. It proposes to utilize current image generation model to reconstruct source and target profiles from embedded features. The authors did thorough experiments and prove quantitatively that the proposed DiffusionFake method outperforms previous baselines.

Strengths

- The problem definition is clear and well-motivated. The proposed solution is also quite simple and intuitive. - All the figures and diagrams are well-designed. Specifically, the first few figures explain the pipeline clearly. Figure 4 and 5 visualize the results in a nice way. - Quantitative results show that the proposed method significantly outperform baselines.

Weaknesses

- Field of application is limited to face swapping with two identities, while claiming a "DeepFake detector". - The Related Work section is not very clear in connecting the current work with previous ones. See questions. - Writing quality is in general good. It might benefit from omitting some details (how many layers?) and provide more intuitive explanation (why this design works?) - Experiments are thorough, but only on two detection metrics. It would be great to instead see something measuring reconstruction quality.

Questions

- Why DFD dataset's best performance happens with a different model architecture than all others? Is there anything special about this dataset and its performance? - In Figure 3A, even with training samples, we can see that the expression is not quite well reconstructed. The target GT expression (mouth slightly open) is "recovered" as source expression. Is this a common observation? How well does the model preserve expression? - Using AUC and EER as metrics are good (and classical) regarding this task. However, as in the ablation study, it would be interesting to see some numerical metrics measuring the reconstruction quality. Is this quality highly-correlated with the detection accuracy? - Is this the first ever work that takes advantage of generative model for face forgery detection? What are some of the most important baselines? Qualitatively how does the current methods outperform the baselines? (i.e. what are some characteristics of those challenging cases that baselines cannot do but this approach can?)

Rating

7

Confidence

3

Soundness

2

Presentation

3

Contribution

3

Limitations

The authors already mentioned a list of limitations at the end of their paper. - Not applicable to blending of multiple identities (>2) or partial manipulation. - Require paired images, not self-supervised. - Could be used as a discriminator to develop harder to distinguish deepfake images. And also two more: - The dataset they use might have limited diversity -- looks like all the examples in the paper are white. A bit concerned about the generalizability to all ethical groups, genders, ages, etc. - The word "Deepfake" might have a slightly broader meaning than face swapping these days. This paper is only detecting face swapping, not "generated faces" in general. It might worth being a bit conservation in claiming the contribution.

Authorsrebuttal2024-08-10

Dear Reviewer tkET, We are deeply grateful for your thorough review and insightful comments on our paper. Your positive evaluation has been highly encouraging. We also appreciate your constructive suggestions, which have significantly improved our paper's depth and completeness. As the discussion period draws to a close, we eagerly await your thoughts on our response. We sincerely hope that our revisions align with your expectations. If there are any remaining concerns or aspects that require clarification, we are ready to address them promptly. Best regards, The Authors

Authorsrebuttal2024-08-12

We sincerely appreciate your thoughtful comments regarding the ethical considerations of our work. We recognize the importance of addressing these concerns in AI research, particularly in the field of DeepFake detection. In response to your points, I would like to share the following thoughts and responses: Our research aims to protect digital integrity by developing methods to detect DeepFake content. This work directly addresses ethical concerns surrounding AI-generated face-swapped images, contributing to the field of privacy protection in the digital age. To validate our method, we utilized widely-accepted public datasets including FaceForensics++, Celeb-DF, DeepFake Detection, DFDC Preview, WildDeepfake, and DiffSwap. These datasets are extensively used in the DeepFake detection domain, with cumulative citations numbering in the thousands. Moreover, these datasets were published in top-tier computer vision conferences, indicating that they have undergone rigorous ethical reviews as part of the conference submission process. These works have provided a solid foundation for deepfake detection research. All datasets used in our study are bound by strict licensing terms that limit their use to non-commercial research and educational purposes. For instance, the DFDC-Preview dataset was created by META using with "paid actors who entered into an agreement to the use and manipulation of their likenesses in our creation of the dataset". FaceForensics++, DeepFake Detection, and DiffSwap stipulate: "Researcher shall use the Database only for non-commercial research and educational purposes." Similarly, Celeb-DF and WildDeepfake have comparable restrictions such as "Our dataset is used only for research purposes, we only release the face sequence rather than the whole video". These licensing terms serve to safeguard privacy and ensure ethical use of the data. These licenses are publicly available on the respective official GitHub repositories. While we cannot provide direct links due to NeurIPS submission restrictions, **we commit to prominently referencing these licenses in our revised paper**. Access to these datasets requires a rigorous application process. We have **obtained proper authorization for each dataset used in our study**, as evidenced by official emails granting us permission to use the data. In our research, we used these datasets solely for simulating and validating our method's effectiveness, without exposing or publicizing any privacy-related content from the datasets. Looking forward, we commit to exploring alternative approaches to further mitigate ethical concerns, including synthetic data generation, privacy-preserving techniques like differential privacy, and advanced data augmentation methods such as random patching, which can improve model performance without exposing identifying information. Based on your valuable feedback, we will add a dedicated section in our paper discussing the ethical considerations of using these datasets. This section will outline our measures to ensure ethical usage, contextualize our work's contribution to ethical AI development, and address the challenges and limitations of current datasets. We believe these additions will strengthen the ethical foundation of our research and contribute to the broader discussion on responsible AI in DeepFake detection. Moreover, in our revised paper, we will ensure proper privacy protection for any displayed facial images, avoiding the exposure of specific facial information. Thank you again for your insightful comments, which have undoubtedly strengthened the ethical foundation of our research. As researchers in this field, we are committed to advancing the technology while being mindful of the ethical implications of our work.

Reviewer tkET2024-08-14

I'm happy with the additional clarifications made by the authors. I'm in general satisfied with this paper and raised my score.

Authorsrebuttal2024-08-14

We are deeply grateful for your reconsideration and the increased score for our work. Thank you for your time, expertise, and the constructive dialogue throughout this review process.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC