arXiv · 2505.19442
Style2Code: A Style-Controllable Code Generation Framework with Dual-Modal Contrastive Representation Learning
Abstract
Controllable code generation, the ability to synthesize code that follows a specified style while maintaining functionality, remains a challenging task. We propose a two-stage training framework combining contrastive learning and conditional decoding to enable flexible style control. The first stage aligns code style representations with semantic and structural features. In the second stage, we fine-tune a language model (e.g., Flan-T5) conditioned on the learned style vector to guide generation. Our method supports style interpolation and user personalization via lightweight mixing. Compared to prior work, our unified framework offers improved stylistic control without sacrificing code correctness. This is among the first approaches to combine contrastive alignment with conditional decoding for style-guided code generation.
Explore related subjects
Keep this discovery
Dutao Zhang, Nicolas Rafael Arroyo Arias, YuLong He, Sergey Kovalchuk. 2025-05-26. Style2Code: A Style-Controllable Code Generation Framework with Dual-Modal Contrastive Representation Learning. https://arxiv.org/abs/2505.19442
Cite the original work for its findings. Save a collection to share your selection of sources.