Leveraging Machine Learning and Large Language Models for Automated Image Clustering and Description in Legal Discovery

TL;DR

This paper explores automated image clustering and description generation using machine learning and large language models to improve efficiency in legal discovery workflows involving large-scale image datasets.

Contribution

It systematically evaluates sampling, prompting, and description methods, demonstrating effective strategies for scalable, accurate image cluster descriptions with reduced computational costs.

Findings

Sampling 20 images per cluster maintains quality while reducing costs

LLM-based descriptions outperform traditional TF-IDF methods

Standard prompts outperform chain-of-thought prompts in this context

Abstract

The rapid increase in digital image creation and retention presents substantial challenges during legal discovery, digital archive, and content management. Corporations and legal teams must organize, analyze, and extract meaningful insights from large image collections under strict time pressures, making manual review impractical and costly. These demands have intensified interest in automated methods that can efficiently organize and describe large-scale image datasets. This paper presents a systematic investigation of automated cluster description generation through the integration of image clustering, image captioning, and large language models (LLMs). We apply K-means clustering to group images into 20 visually coherent clusters and generate base captions using the Azure AI Vision API. We then evaluate three critical dimensions of the cluster description process: (1) image sampling strategies, comparing random, centroid-based, stratified, hybrid, and density-based sampling against using all cluster images; (2) prompting techniques, contrasting standard prompting with chain-of-thought prompting; and (3) description generation methods, comparing LLM-based generation with traditional TF-IDF and template-based approaches. We assess description quality using semantic similarity and coverage metrics. Results show that strategic sampling with 20 images per cluster performs comparably to exhaustive inclusion while significantly reducing computational cost, with only stratified sampling showing modest degradation. LLM-based methods consistently outperform TF-IDF baselines, and standard prompts outperform chain-of-thought prompts for this task. These findings provide practical guidance for deploying scalable, accurate cluster description systems that support high-volume workflows in legal discovery and other domains requiring automated organization of large image collections.