TY - GEN
T1 - Automating Generation of UML Sequence Diagrams Using LLM
AU - Hari, Gayathri
AU - Geetha, M.
N1 - Publisher Copyright:
© The Author(s), under exclusive license to Springer Nature Switzerland AG 2026.
PY - 2026
Y1 - 2026
N2 - Manually creating UML sequence diagrams from Software Requirements Specification (SRS) papers takes a lot of effort and is prone to errors. Traditional rule-based NLP methods struggle with ambiguity and require extensive human supervision. Large language models (LLMs) and graph-based semantic retrieval using Neo4j are combined in the hybrid Graph Retrieval-Augmented Generation (GraphRAG) architecture shown in this study. Our approach improves accuracy by embedding extracted UML components in a graph and utilizing retrieval-enhanced prompting for guided diagram production. Our technique was tested using four different generating strategies: (i) no prompting or RAG, (ii) only prompting, (iii) retrieval based on FAISS, and (iv) the proposed Neo4j-based retrieval. Across five domain-specific SRS documents, the Neo4j-enhanced solution achieves the highest levels of semantic alignment, structural completeness, and overall accuracy, outperforming all baselines. We also introduce a multi-metric evaluation methodology created especially for the quality of sequence diagrams. These results show that structured retrieval is a viable method for developing AI-assisted software modeling.
AB - Manually creating UML sequence diagrams from Software Requirements Specification (SRS) papers takes a lot of effort and is prone to errors. Traditional rule-based NLP methods struggle with ambiguity and require extensive human supervision. Large language models (LLMs) and graph-based semantic retrieval using Neo4j are combined in the hybrid Graph Retrieval-Augmented Generation (GraphRAG) architecture shown in this study. Our approach improves accuracy by embedding extracted UML components in a graph and utilizing retrieval-enhanced prompting for guided diagram production. Our technique was tested using four different generating strategies: (i) no prompting or RAG, (ii) only prompting, (iii) retrieval based on FAISS, and (iv) the proposed Neo4j-based retrieval. Across five domain-specific SRS documents, the Neo4j-enhanced solution achieves the highest levels of semantic alignment, structural completeness, and overall accuracy, outperforming all baselines. We also introduce a multi-metric evaluation methodology created especially for the quality of sequence diagrams. These results show that structured retrieval is a viable method for developing AI-assisted software modeling.
UR - https://www.scopus.com/pages/publications/105039644773
UR - https://www.scopus.com/pages/publications/105039644773#tab=citedBy
U2 - 10.1007/978-3-032-19185-4_5
DO - 10.1007/978-3-032-19185-4_5
M3 - Conference contribution
AN - SCOPUS:105039644773
SN - 9783032191847
T3 - Lecture Notes in Networks and Systems
SP - 55
EP - 66
BT - Artificial Intelligence
A2 - Chakravorty, Antorweep
A2 - Hussain, Shahid
A2 - Kumari, Rajani
PB - Springer Science and Business Media Deutschland GmbH
T2 - 3rd International Conference on Artificial Intelligence: Theory and Applications, AITA 2025
Y2 - 1 August 2025 through 2 August 2025
ER -