Context-Aware Semantic Segmentation: Enhancing Pixel-Level Understanding with Large Language Models for Advanced Vision Applications
Journal:
arXiv
Published Date:
Mar 25, 2025
Abstract
Semantic segmentation has made significant strides in pixel-level image
understanding, yet it remains limited in capturing contextual and semantic
relationships between objects. Current models, such as CNN and
Transformer-based architectures, excel at identifying pixel-level features but
fail to distinguish semantically similar objects (e.g., "doctor" vs. "nurse" in
a hospital scene) or understand complex contextual scenarios (e.g.,
differentiating a running child from a regular pedestrian in autonomous
driving). To address these limitations, we proposed a novel Context-Aware
Semantic Segmentation framework that integrates Large Language Models (LLMs)
with state-of-the-art vision backbones. Our hybrid model leverages the Swin
Transformer for robust visual feature extraction and GPT-4 for enriching
semantic understanding through text embeddings. A Cross-Attention Mechanism is
introduced to align vision and language features, enabling the model to reason
about context more effectively. Additionally, Graph Neural Networks (GNNs) are
employed to model object relationships within the scene, capturing dependencies
that are overlooked by traditional models. Experimental results on benchmark
datasets (e.g., COCO, Cityscapes) demonstrate that our approach outperforms the
existing methods in both pixel-level accuracy (mIoU) and contextual
understanding (mAP). This work bridges the gap between vision and language,
paving the path for more intelligent and context-aware vision systems in
applications including autonomous driving, medical imaging, and robotics.