Skip to main navigation Skip to search Skip to main content

Automating Generation of UML Sequence Diagrams Using LLM

  • Gayathri Hari*
  • , M. Geetha
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

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.

Original languageEnglish
Title of host publicationArtificial Intelligence
Subtitle of host publicationTheory and Applications - Proceedings of AITA 2025
EditorsAntorweep Chakravorty, Shahid Hussain, Rajani Kumari
PublisherSpringer Science and Business Media Deutschland GmbH
Pages55-66
Number of pages12
ISBN (Print)9783032191847
DOIs
Publication statusPublished - 2026
Externally publishedYes
Event3rd International Conference on Artificial Intelligence: Theory and Applications, AITA 2025 - Bangalore, India
Duration: 01-08-202502-08-2025

Publication series

NameLecture Notes in Networks and Systems
Volume1865 LNNS
ISSN (Print)2367-3370
ISSN (Electronic)2367-3389

Conference

Conference3rd International Conference on Artificial Intelligence: Theory and Applications, AITA 2025
Country/TerritoryIndia
CityBangalore
Period01-08-2502-08-25

All Science Journal Classification (ASJC) codes

  • Control and Systems Engineering
  • Signal Processing
  • Computer Networks and Communications

Fingerprint

Dive into the research topics of 'Automating Generation of UML Sequence Diagrams Using LLM'. Together they form a unique fingerprint.

Cite this