Integrating Large Language Models and Graph Convolutional Networks for Semi-Supervised Image Classification

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision, Natural Language Processing · Depth: Advanced, quick

Summary

A novel method integrates Large Language Models (LLMs) and Graph Convolutional Networks (GCNs) to enhance semi-supervised image classification, addressing the challenge of constructing effective graphs for visual data. Unlike traditional approaches that rely on kNN or reciprocal kNN algorithms applied to feature vectors from deep learning backbones, this technique employs a Vision Language Model (VLM) to generate textual descriptions for images. Subsequently, an LLM processes these descriptions to estimate semantic similarity scores between connected images. These scores are then utilized to prune edges in the initial kNN and reciprocal kNN graphs, effectively filtering out semantically irrelevant neighbors. Experimental results demonstrate that this LLM-guided graph refinement significantly improves classification accuracy, particularly when applied to kNN graphs and specific deep learning backbones. The source code is publicly available.

Key takeaway

For Machine Learning Engineers developing semi-supervised image classification systems with Graph Convolutional Networks, this research indicates a valuable strategy. If your current GCN performance is limited by graph construction, consider integrating Vision Language Models to generate image descriptions and then using Large Language Models to semantically prune kNN or reciprocal kNN graphs. This approach can significantly improve classification accuracy by filtering out irrelevant connections, offering a novel path to enhance model robustness with unlabeled data.

Key insights

Integrating LLMs with VLMs to semantically prune GCN image graphs improves semi-supervised classification accuracy.

Principles

Method

A VLM generates image descriptions, which an LLM processes to estimate semantic similarity scores. These scores then guide the pruning of kNN and reciprocal kNN graph edges, filtering semantically irrelevant neighbors.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.