Proceedings

MA-RAG: Automating Role Engineering for RESTful APIs with Multi-Head Attention and Retrieval-Augmented Generation

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 7607-7615.

https://doi.org/10.24963/ijcai.2025/846

Abstract

This paper addresses the role engineering problem for RESTful applications and proposes a role engineering method based on multi-head attention and Retrieval Augmented Generation called MA-RAG. The method first performs fine-grained control flow analysis on the system source code to extract permission information of API handlers. Then, using basic blocks as units, it employs pre-trained code models to convert the source code into semantic vectors, which are stored in the retrieval augmented generation model. On this basis, a call chain structure tree is constructed with permissions as the center, utilizing the multi-head attention mechanism to aggregate semantic information of different code granularities from bottom to top, with each attention head corresponding to a role engineering objective. Finally, the root vectors of each permission tree are subjected to self-supervised clustering to adaptively determine the number of roles and perform division. We evaluated MA-RAG on 284 real-world software systems, and the results show that compared with other methods, MA-RAG can significantly save time overhead, reduce the number of generated roles, lower the role permission overlap rate, and improve the interpretability score.

  1. [Abolfathi et al., 2021] Masoumeh Abolfathi, Zohreh Raghebi, Haadi Jafarian, and Farnoush Banaei-Kashani. A scalable role mining approach for large organizations. In Proceedings of the 2021 ACM Workshop on Security and Privacy Analytics, pages 45-54, 2021.
  2. [Aloise et al., 2009] Daniel Aloise, Amit Deshpande, Pierre Hansen, and Preyas Popat. Np-hardness of euclidean sum- of-squares clustering. Machine learning, 75:245-248, 2009.
  3. [Anderer et al., 2021] Simon Anderer, Bernd Scheuermann, Sanaz Mostaghim, Patrick Bauerle, and Matthias Beil. Rmplib: a library of benchmarks for the role mining prob- lem. In Proceedings of the 26th ACM Symposium on Ac- cess Control Models and Technologies, pages 3-13, 2021.
  4. [Anderer et al., 2022] Simon Anderer, Falk Schrader, Bernd Scheuermann, and Sanaz Mostaghim. Evolutionary algo- rithms for the constrained two-level role mining problem. In European Conference on Evolutionary Computation in Combinatorial Optimization (Part of EvoStar), pages 79- 94. Springer, 2022.
  5. [Anderer et al., 2023] Simon Anderer, Nicolas Justen, Bernd Scheuermann, and Sanaz Mostaghim. Interactive role min- ing including expert knowledge into evolutionary algo- rithms. In IJCCI, pages 151-162, 2023.
  6. [Blundo et al., 2023] Carlo Blundo, Stelvio Cimato, and Luisa Siniscalchi. Heuristics for constrained role mining in the post-processing framework. Journal of Ambient In- telligence and Humanized Computing, 14(8):9925-9937, 2023.
  7. [Chen et al., 2022] Wan Chen, Daojun Han, Lei Zhang, Qi Xiao, Qiuyue Li, and Hongzhen Xiang. A model study on hierarchical assisted exploration of rbac. International Journal of Digital Crime and Forensics (IJDCF), 14(2):1- 13, 2022.
  8. [Coyne, 1996] Edward J Coyne. Role engineering. In Pro- ceedings of the first ACM Workshop on Role-based access control, page 4. ACM, 1996.
  9. [Crampton et al., 2022] Jason Crampton, Eduard Eiben, Gregory Gutin, Daniel Karapetyan, and Diptapriyo Ma- jumdar. Generalized noise role mining. In Proceedings of the 27th ACM on Symposium on Access Control Models and Technologies, pages 91-102, 2022.
  10. [Du et al., 2022] Yajun Du, Yakun Wang, Jinrong Hu, Xi- anyong Li, and Xiaoliang Chen. An emotion role mining approach based on multiview ensemble learning in social networks. Information fusion, 88:100-114, 2022.
  11. [Epstein and Sandhu, 2001] Pete Epstein and Ravi Sandhu. Engineering of role/permission assignments. In Computer Security Applications Conference, 2001. ACSAC 2001. Proceedings 17th Annual, pages 127-136. IEEE, 2001.
  12. [Ferrante et al., 1987] Jeanne Ferrante, Karl J Ottenstein, and Joe D Warren. The program dependence graph and its use in optimization. ACM Transactions on Programming Languages and Systems (TOPLAS), 9(3):319-349, 1987.
  13. [Fielding, 2000] Roy Thomas Fielding. Architectural styles and the design of network-based software architectures. PhD thesis, University of California, Irvine, 2000.
  14. [Guo and Tripunitara, 2022] Qiang Guo and Mahesh Tripunitara. The secrecy resilience of access control policies and its application to role mining. In Proceedings of the 27th ACM on Symposium on Access Control Models and Technologies, pages 115-126, 2022.
  15. [Kuhlmann et al., 2003] Martin Kuhlmann, Dalia Shohat, and Gerhard Schimpf. Role mining-revealing business roles for security administration using data mining tech- nology. In Proceedings of the eighth ACM symposium on Access control models and technologies, pages 179-186. ACM, 2003.
  16. [Liu et al., 2022] Ye Liu, Yi Li, Shang-Wei Lin, and Cyrille Artho. Finding permission bugs in smart contracts with role mining. In Proceedings of the 31st ACM SIGSOFT International Symposium on Software Testing and Analy- sis, pages 716-727, 2022.
  17. [Neumann and Strembeck, 2002] Gustaf Neumann and Mark Strembeck. A scenario-driven role engineering process for functional rbac roles. In Proceedings of the seventh ACM symposium on Access control models and technologies, pages 33-42. ACM, 2002.
  18. [OpenStack, 2024] OpenStack. Openstack, the open-source cloud platform. https://www.openstack.org, 2024. Link
  19. [Pilipchuk et al., 2021] Roman Pilipchuk, Robert Heinrich, and Ralf H Reussner. Automatically extracting business level access control requirements from bpmn models to align rbac policies. In ICISSP, pages 300-307, 2021.
  20. [Rashid et al., 2021] Aqsa Rashid, Asif Masood, and Haider Abbas. Cryptographic framework for role control remedy: A secure role engineering mechanism for single author- ity organizations. Future Generation Computer Systems, 117:245-258, 2021.
  21. [Saltzer and Schroeder, 1975] Jerome H Saltzer and Michael D Schroeder. The protection of informa- tion in computer systems. Proceedings of the IEEE, 63(9):1278-1308, 1975.
  22. [Sandhu, 1998] Ravi S Sandhu. Role-based access control. In Advances in computers, volume 46, pages 237-286.
  23. [Vaidya et al., 2007] Jaideep Vaidya, Vijayalakshmi Atluri, and Qi Guo. The role mining problem: finding a mini- mal descriptive set of roles. In Proceedings of the 12th ACM symposium on Access control models and technolo- gies, pages 175-184. ACM, 2007.
  24. [Vaidya et al., 2008] Jaideep Vaidya, Vijayalakshmi Atluri, Qi Guo, and Nabil Adam. Migrating to optimal rbac with minimal perturbation. In Proceedings of the 13th ACM symposium on Access control models and technologies, pages 11-20. ACM, 2008.
  25. [Vaswani et al., 2017] Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Lukasz Kaiser, and Illia Polosukhin. Attention is all you need.(nips), 2017. arXiv preprint arXiv:1706.03762, 10:S0140525X16001837, 2017.
  26. [Xia et al., 2023] Yutang Xia, Yang Luo, Wu Luo, Qingni Shen, Yahui Yang, and Zhonghai Wu. A role engineering approach based on spectral clustering analysis for restful permissions in cloud. In ICASSP 2023-2023 IEEE Inter- national Conference on Acoustics, Speech and Signal Pro- cessing (ICASSP), pages 1-5. IEEE, 2023.
  27. [Zhang et al., 2007] Dana Zhang, Kotagiri Ramamohanarao, and Tim Ebringer. Role engineering using graph optimi- sation. In Proceedings of the 12th ACM symposium on Access control models and technologies, pages 139-144. ACM, 2007.

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

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