Proceedings

SemanticMask: A Contrastive View Design for Anomaly Detection in Tabular Data

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 2370-2378.

https://doi.org/10.24963/ijcai.2024/262

Abstract

Contrastive learning based on data augmentation techniques has recently achieved substantial advancement in learning a representation well-suited for anomaly detection in image domain. However, due to the lack of spatial structure, designing effective data augmentation methods for tabular data remains challenging. Conventional techniques, such as random mask, disregard the inter-feature correlations and fail to accurately represent the data. To address this issue, we propose a novel augmentation technique called SemanticMask which leverages the semantic information from column names to generate better augmented views. SemanticMask aims to ensure that the shared information between views contains sufficient information for anomaly detection without redundancy. We analyze the relationship between shared information and anomaly detection performance and empirically demonstrate that good views for tabular anomaly detection tasks are feature-dependent. Our experiment results validate the superiority of SemanticMask over the state-of-the-art anomaly detection methods and existing augmentation techniques for tabular data. In further evaluations of the multi-class novelty detection task, SemanticMask also significantly outperforms the baseline.

  1. [Bahri et al., 2021] Dara Bahri, Heinrich Jiang, Yi Tay, and Donald Metzler. Scarf: Self-supervised contrastive learn- ing using random feature corruption. In Proceedings of the International Conference on Learning Representa- tions, 2021.
  2. [Breunig et al., 2000] Markus M Breunig, Hans-Peter Kriegel, Raymond T Ng, and Jörg Sander. Lof: iden- tifying density-based local outliers. In Proceedings of the 2000 ACM SIGMOD international conference on Management of data, pages 93-104, 2000.
  3. [Chandola et al., 2009] Varun Chandola, Arindam Banerjee, and Vipin Kumar. Anomaly detection: A survey. ACM computing surveys (CSUR), 41(3):1-58, 2009.
  4. [Chen et al., 2020] Ting Chen, Simon Kornblith, Moham- mad Norouzi, and Geoffrey Hinton. A simple framework for contrastive learning of visual representations. In Inter- national Conference on Machine Learning, pages 1597- 1607, 2020.
  5. [Cho et al., 2021] Hyunsoo Cho, Jinseok Seol, and Sang-goo Lee. Masked contrastive learning for anomaly detection. In Proceedings of the Thirtieth International Joint Confer- ence on Artificial Intelligence, pages 1434-1441, 2021. IJCAI
  6. [Derrac et al., 2015] J Derrac, S Garcia, L Sanchez, and F Herrera. Keel data-mining software tool: Data set repos- itory, integration of algorithms and experimental analysis framework. Journal of Multiple-Valued Logic & Soft Com- puting, 17, 2015.
  7. [Eskin, 2000] Eleazar Eskin. Anomaly detection over noisy data using learned probability distributions. In Interna- tional Conference on Machine Learning, pages 255-262, 2000.
  8. [Golan and El-Yaniv, 2018] Izhak Golan and Ran El-Yaniv. Deep anomaly detection using geometric transformations. In Advances in Neural Information Processing Systems, volume 31, pages 9781-9791, 2018.
  9. [Han et al., 2022] Songqiao Han, Xiyang Hu, Hailiang Huang, Minqi Jiang, and Yue Zhao. Adbench: Anomaly detection benchmark. In Advances in Neural Informa- tion Processing Systems, volume 35, pages 32142-32159, 2022.
  10. [Hartigan et al., 1979] John A Hartigan, Manchek A Wong, et al. A k-means clustering algorithm. Journal of the royal statistical society. series c (applied statistics), 28(1):100- 108, 1979.
  11. [He et al., 2020] Kaiming He, Haoqi Fan, Yuxin Wu, Sain- ing Xie, and Ross Girshick. Momentum contrast for un- supervised visual representation learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pat- tern Recognition, pages 9726-9735, 2020.
  12. [Khosla et al., 2020] Prannay Khosla, Piotr Teterwak, Chen Wang, Aaron Sarna, Yonglong Tian, Phillip Isola, Aaron Maschinot, Ce Liu, and Dilip Krishnan. Supervised con- trastive learning. In Advances in Neural Information Pro- cessing Systems, volume 33, pages 18661-18673, 2020.
  13. [Lazarevic and Kumar, 2005] Aleksandar Lazarevic and Vipin Kumar. Feature bagging for outlier detection. In Proceedings of the Eleventh ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pages 157-166, 2005.
  14. [Li et al., 2020] Zheng Li, Yue Zhao, Nicola Botta, Cezar Ionescu, and Xiyang Hu. Copod: copula-based outlier de- tection. In 2020 IEEE International Conference on Data Mining, pages 1118-1123, 2020.
  15. [Li et al., 2021] Chun-Liang Li, Kihyuk Sohn, Jinsung Yoon, and Tomas Pfister. Cutpaste: Self-supervised learn- ing for anomaly detection and localization. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pat- tern Recognition, pages 9664-9674, 2021.
  16. [Li et al., 2022] Zheng Li, Yue Zhao, Xiyang Hu, Nicola Botta, Cezar Ionescu, and George Chen. Ecod: Unsuper- vised outlier detection using empirical cumulative distri- bution functions. IEEE Transactions on Knowledge and Data Engineering, pages 1-13, 2022.
  17. [Markelle et al., 2013] Kelly Markelle, Longjohn Rachel, and Nottingham Kolby. The uci machine learning reposi- tory. https://archive.ics.uci.edu, 2013. Accessed: 2023-6- 28. Link
  18. [Nguyen et al., 2019] Duc Tam Nguyen, Zhongyu Lou, Michael Klar, and Thomas Brox. Anomaly detection with multiple-hypotheses predictions. In International Confer- ence on Machine Learning, volume 35, pages 4800-4809, 2019.
  19. [Oord et al., 2018] Aaron van den Oord, Yazhe Li, and Oriol Vinyals. Representation learning with contrastive predic- tive coding, 2018. [OpenAI, 2023] OpenAI. Gpt-4 technical report, 2023.
  20. [Peng et al., 2022] Xiangyu Peng, Kai Wang, Zheng Zhu, Mang Wang, and Yang You. Crafting better contrastive views for siamese representation learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pat- tern Recognition, pages 16010-16019, 2022.
  21. [Pidhorskyi et al., 2018] Stanislav Pidhorskyi, Ranya Al- mohsen, and Gianfranco Doretto. Generative probabilis- tic novelty detection with adversarial autoencoders. In Advances in Neural Information Processing Systems, vol- ume 31, pages 6823-6834, 2018.
  22. [Rayana, 2016] Shebuti Rayana. Odds library. http://odds. cs.stonybrook.edu, 2016. Accessed: 2023-3-28. Link
  23. [Reimers and Gurevych, 2019] Nils Reimers and Iryna Gurevych. Sentence-bert: Sentence embeddings using siamese bert-networks. In Proceedings of the Conference on Empirical Methods in Natural Language Processing, pages 3980-3990, 2019.
  24. [Reiss et al., 2021] Tal Reiss, Niv Cohen, Liron Bergman, and Yedid Hoshen. Panda: Adapting pretrained features for anomaly detection and segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pat- tern Recognition, pages 2806-2814, 2021.
  25. [Ruff et al., 2018] Lukas Ruff, Robert Vandermeulen, Nico Goernitz, Lucas Deecke, Shoaib Ahmed Siddiqui, Alexan- der Binder, Emmanuel Müller, and Marius Kloft. Deep one-class classification. In International Conference on Machine Learning, pages 4390-4399, 2018.
  26. [Ruff et al., 2021] Lukas Ruff, Jacob R Kauffmann, Robert A Vandermeulen, Grégoire Montavon, Woj- ciech Samek, Marius Kloft, Thomas G Dietterich, and Klaus-Robert Müller. A unifying review of deep and shallow anomaly detection. Proceedings of the IEEE, 109(5):756-795, 2021.
  27. [Schölkopf et al., 1999] Bernhard Schölkopf, Robert C Williamson, Alex Smola, John Shawe-Taylor, and John Platt. Support vector method for novelty detection. In Advances in Neural Information Processing Systems, volume 12, pages 582-588, 1999.
  28. [Schölkopf et al., 2001] Bernhard Schölkopf, John C Platt, John Shawe-Taylor, Alex J Smola, and Robert C Williamson. Estimating the support of a high-dimensional distribution. Neural computation, 13(7):1443-1471, 2001.
  29. [Sehwag et al., 2021] Vikash Sehwag, Mung Chiang, and Prateek Mittal. Ssd: A unified framework for self- supervised outlier detection. In Proceedings of the Inter- national Conference on Learning Representations, 2021.
  30. [Shenkar and Wolf, 2022] Tom Shenkar and Lior Wolf. Anomaly detection for tabular data with internal con- trastive learning. In Proceedings of the International Con- ference on Learning Representations, 2022.
  31. [Sohn et al., 2021] Kihyuk Sohn, Chun-Liang Li, Jinsung Yoon, Minho Jin, and Tomas Pfister. Learning and eval- uating representations for deep one-class classification. In Proceedings of the International Conference on Learning Representations, 2021.
  32. [Suhara et al., 2022] Yoshihiko Suhara, Jinfeng Li, Yuliang Li, Dan Zhang, C ¸agatay Demiralp, Chen Chen, and Wang- Chiew Tan. Annotating columns with pre-trained language models. In Proceedings of the 2022 ACM SIGMOD inter- national conference on Management of Data, pages 1493- 1503, 2022.
  33. [Tack et al., 2020] Jihoon Tack, Sangwoo Mo, Jongheon Jeong, and Jinwoo Shin. Csi: Novelty detection via con- trastive learning on distributionally shifted instances. In Advances in Neural Information Processing Systems, vol- ume 33, pages 11839-11852, 2020.
  34. [Tian et al., 2020a] Yonglong Tian, Dilip Krishnan, and Phillip Isola. Contrastive multiview coding. In European Conference on Computer Vision, pages 776-794, 2020.
  35. [Tian et al., 2020b] Yonglong Tian, Chen Sun, Ben Poole, Dilip Krishnan, Cordelia Schmid, and Phillip Isola. What makes for good views for contrastive learning? In Ad- vances in Neural Information Processing Systems, vol- ume 33, pages 6827-6839, 2020.
  36. [Verma et al., 2021] Vikas Verma, Thang Luong, Kenji Kawaguchi, Hieu Pham, and Quoc Le. Towards domain- agnostic contrastive learning. In International Conference on Machine Learning, pages 10530-10541, 2021.
  37. [Wang and Cherian, 2019] Jue Wang and Anoop Cherian. Gods: Generalized one-class discriminative subspaces for anomaly detection. In Proceedings of the IEEE Interna- tional Conference on Computer Vision, pages 8200-8210, 2019.
  38. [Wang and Sun, 2022] Zifeng Wang and Jimeng Sun. Transtab: Learning transferable tabular transformers across tables. In Advances in Neural Information Processing Systems, volume 35, pages 2902-2915, 2022.
  39. [Wang et al., 2022] Haoqing Wang, Xun Guo, Zhi-Hong Deng, and Yan Lu. Rethinking minimal sufficient repre- sentation in contrastive learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 16020-16029, 2022.
  40. [Wen and Li, 2021] Zixin Wen and Yuanzhi Li. Toward un- derstanding the feature learning process of self-supervised contrastive learning. In International Conference on Ma- chine Learning, pages 11112-11122, 2021.
  41. [Xiang et al., 2023] Haolong Xiang, Xuyun Zhang, Hong- sheng Hu, Lianyong Qi, Wanchun Dou, Mark Dras, Amin Beheshti, and Xiaolong Xu. Optiforest: Optimal isola- tion forest for anomaly detection. In Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence, pages 2379-2387, 2023. IJCAI
  42. [Xie et al., 2022] Xu Xie, Fei Sun, Zhaoyang Liu, Shiwen Wu, Jinyang Gao, Jiandong Zhang, Bolin Ding, and Bin Cui. Contrastive learning for sequential recommendation. In International Conference on Data Engineering, pages 1259-1273, 2022.
  43. [Yan et al., 2021] Xudong Yan, Huaidong Zhang, Xuemiao Xu, Xiaowei Hu, and Pheng-Ann Heng. Learning seman- tic context from normal samples for unsupervised anomaly detection. In Proceedings of the AAAI Conference on Ar- tificial Intelligence, pages 3110-3118, 2021.
  44. [Yang et al., 2021] Jingkang Yang, Kaiyang Zhou, Yixuan Li, and Ziwei Liu. Generalized out-of-distribution detec- tion: A survey, 2021.
  45. [Yoon et al., 2020] Jinsung Yoon, Yao Zhang, James Jordon, and Mihaela van der Schaar. Vime: Extending the success of self-and semi-supervised learning to tabular domain. In Advances in Neural Information Processing Systems, vol- ume 33, pages 11033-11043, 2020.
  46. [Zhai et al., 2016] Shuangfei Zhai, Yu Cheng, Weining Lu, and Zhongfei Zhang. Deep structured energy based mod- els for anomaly detection. In International Conference on Machine Learning, pages 1100-1109, 2016.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards