Scopus

Automated UML Generation: A Framework for Class Diagram Synthesis and Multimodal Validation

Năm XB 2025 Tạp chí / Hội thảo International Conference on Future Data and Security Engineering (FDSE2025) 212-224 DOI / Link https://doi.org/10.1007/978-981-95-4724-1_15 ↗

Tác giả

Van-Viet Nguyen ^✉ ; Huu-Khanh Nguyen ; Kim-Son Nguyen ; Thi Minh-Hue Luong ; Duc-Quang Vu ; The-Vinh Nguyen

Tóm tắt

The complexity of modern software systems necessitates robust modeling tools, with UML Class Diagrams serving as a cornerstone for representing static system architecture. However, manual creation of these diagrams is a significant bottleneck, being both time-intensive and prone to error. This paper extends our framework for automated UML generation to tackle the intricate challenge of Class Diagram synthesis. We propose a dual-model pipeline where a lightweight LLM (LLaMA 3.2 1B-Instruct) generates detailed technical specifications, which are then translated into PlantUML Class Diagram code by a powerful reasoning model (DeepSeek-R1-Distill-Qwen-32B). This process yielded a novel dataset of 5,000 samples, each comprising a technical specification, a PlantUML code block, and a corresponding diagram. To ensure architectural integrity, we deploy an automated multimodal validation system using …

Tài liệu tham khảo

[1] Ambler, S.W.: The object primer: agile model-driven development with UML 2.0. Cambridge University Press (2004)

[2] Bai, J., et al.: Qwen-vl: a versatile vision-language model for understanding, localization, text reading, and beyond. arXiv preprint arXiv:2308.12966 (2023)

[3] Belady, L.A., Lehman, M.: Program evolution: processes of software change. Academic Press (1985)

[4] Booch, G.: The unified modeling language user guide. Pearson Education India (2005)

[5] Chen, Y.-C., et al.: UNITER: UNiversal Image-TExt Representation learning. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.-M. (eds.) ECCV 2020. LNCS, vol. 12375, pp. 104–120. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58577-8_7

[6] Conrardy, A., Cabot, J.: From image to UML: first results of image based UML diagram generation using LLMs. arXiv preprint arXiv:2404.11376 (2024)

[7] DeepSeek-AI: Deepseek-r1: Incentivizing reasoning capability in LLMs via reinforcement learning (2025). https://arxiv.org/abs/2501.12948

[8] Lu, J., Batra, D., Parikh, D., Lee, S.: Vilbert: pretraining task-agnostic visiolinguistic representations for vision-and-language tasks. In: Advances in Neural Information Processing Systems, vol. 32 (2019)

[9] Tan, H., Bansal, M.: Lxmert: learning cross-modality encoder representations from transformers. arXiv preprint arXiv:1908.07490 (2019)

[10] Touvron, H., et al.: Llama: open and efficient foundation language models (2023). https://arxiv.org/abs/2302.13971

[11] Üstün, A., et al.: Aya model: an instruction finetuned open-access multilingual language model. arXiv preprint arXiv:2402.07827 (2024)

[12] Viet, N.V.: Umlcode-classdiagram-deepseek-32b-scored (revision b93c8b0) (2025). https://doi.org/10.57967/hf/5932. https://huggingface.co/datasets/nguyenvanviet/UMLCode-ClassDiagram-DeepSeek-32B-Scored

[13] Viet, N.V., Vinh, N.T.: Large language models in software engineering: a systematic review and vision. J. Educ. Sustain. Innov. 2(2), 146–156 (2024). https://doi.org/10.56916/jesi.v2i2.968. https://ejournal.papanda.org/index.php/jesi/article/view/968

[14] Wang, P., et al.: Qwen2-vl: enhancing vision-language model’s perception of the world at any resolution. arXiv preprint arXiv:2409.12191 (2024)

[15] Yue, X., et al.: Mmmu: a massive multi-discipline multimodal understanding and reasoning benchmark for expert AGI. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 9556–9567 (2024)

[16] Zhang, P., et al.: Vinvl: revisiting visual representations in vision-language models. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 5579–5588 (2021)

[17] Zhou, L., Palangi, H., Zhang, L., Hu, H., Corso, J., Gao, J.: Unified vision-language pre-training for image captioning and VQA. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, pp. 13041–13049 (2020)

← Quay lại danh sách bài báo