Merge pull request #97 from pygod-team/gadnr

Add the GADNR model
pygod-team · Dec 21, 2023 · 9554d2e · 9554d2e
2 parents 00a60dd + 9a55bb3
commit 9554d2e
Show file tree

Hide file tree

Showing 13 changed files with 1,487 additions and 4 deletions.
diff --git a/README.rst b/README.rst
@@ -178,6 +178,7 @@ OCGNN               2021   GNN          Yes          [#Wang2021One]_
 CoLA                2021   GNN+AE+SSL   Yes          [#Liu2021Anomaly]_
 GUIDE               2021   GNN+AE       Yes          [#Yuan2021Higher]_
 CONAD               2022   GNN+AE+SSL   Yes          [#Xu2022Contrastive]_
+GADNR               2023   GNN+AE       Yes          [#Roy2023Gadnr]_
 ==================  =====  ===========  ===========  ========================================
 
 
@@ -253,3 +254,5 @@ Reference
 .. [#Yuan2021Higher] Yuan, X., Zhou, N., Yu, S., Huang, H., Chen, Z. and Xia, F., 2021, December. Higher-order Structure Based Anomaly Detection on Attributed Networks. In 2021 IEEE International Conference on Big Data (Big Data).
 
 .. [#Xu2022Contrastive] Xu, Z., Huang, X., Zhao, Y., Dong, Y., and Li, J., 2022. Contrastive Attributed Network Anomaly Detection with Data Augmentation. In Proceedings of the 26th Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD).
+
+.. [#Roy2023Gadnr] Roy, A., Shu, J., Li, J., Yang, C., Elshocht, O., Smeets, J. and Li, P., 2023. GAD-NR: Graph Anomaly Detection via Neighborhood Reconstruction. In Proceedings of the 17th ACM International Conference on Web Search and Data Mining (WSDM).
diff --git a/docs/index.rst b/docs/index.rst
@@ -104,6 +104,7 @@ OCGNN               2021   GNN          Yes          :class:`pygod.detector.OCGN
 CoLA                2021   GNN+AE+SSL   Yes          :class:`pygod.detector.CoLA`
 GUIDE               2021   GNN+AE       Yes          :class:`pygod.detector.GUIDE`
 CONAD               2022   GNN+AE+SSL   Yes          :class:`pygod.detector.CONAD`
+GADNR               2023   GNN+AE       Yes          :class:`pygod.detector.GADNR`
 ==================  =====  ===========  ===========  ==============================================
 
 

diff --git a/docs/pygod.detector.rst b/docs/pygod.detector.rst
@@ -14,6 +14,7 @@ pygod.detector
     ~pygod.detector.DOMINANT
     ~pygod.detector.DONE
     ~pygod.detector.GAAN
+    ~pygod.detector.GADNR
     ~pygod.detector.GAE
     ~pygod.detector.GUIDE
     ~pygod.detector.OCGNN

diff --git a/docs/pygod.nn.rst b/docs/pygod.nn.rst
@@ -12,6 +12,7 @@ pygod.nn
     ~pygod.nn.DOMINANTBase
     ~pygod.nn.DONEBase
     ~pygod.nn.GAANBase
+    ~pygod.nn.GADNRBase
     ~pygod.nn.GAEBase
     ~pygod.nn.GUIDEBase
     ~pygod.nn.OCGNNBase
diff --git a/docs/zreferences.bib b/docs/zreferences.bib
@@ -171,4 +171,11 @@ @article{kipf2016variational
   author={Kipf, Thomas N and Welling, Max},
   journal={arXiv preprint arXiv:1611.07308},
   year={2016}
-}
+}
+
+@inproceedings{roy2023gadnr,
+  title  = {GAD-NR : Graph Anomaly Detection via Neighborhood Reconstruction},
+  author = {Roy, Amit and Shu, Juan and Li, Jia and Yang, Carl and Elshocht, Olivier and Smeets, Jeroen and Li, Pan},
+  booktitle={Proceedings of the 17th ACM International Conference on Web Search and Data Mining},
+  year   = {2024}
+}
diff --git a/pygod/detector/__init__.py b/pygod/detector/__init__.py
@@ -9,6 +9,7 @@
 from .dominant import DOMINANT
 from .done import DONE
 from .gaan import GAAN
+from .gadnr import GADNR
 from .gae import GAE
 from .guide import GUIDE
 from .ocgnn import OCGNN
@@ -18,6 +19,6 @@
 
 __all__ = [
     "Detector", "DeepDetector", "AdONE", "ANOMALOUS", "AnomalyDAE", "CoLA",
-    "CONAD", "DOMINANT", "DONE", "GAAN", "GAE", "GUIDE", "OCGNN", "ONE",
+    "CONAD", "DOMINANT", "DONE", "GAAN", "GADNR", "GAE", "GUIDE", "OCGNN", "ONE",
     "Radar", "SCAN"
 ]