Contrastive Concept-Tree Search for LLM-Assisted Algorithm Discovery

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Leleu, Timothee, Gunathilaka, Sudeera, Ghimenti, Federico, Ganguli, Surya
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914303851888640
author Leleu, Timothee
Gunathilaka, Sudeera
Ghimenti, Federico
Ganguli, Surya
author_facet Leleu, Timothee
Gunathilaka, Sudeera
Ghimenti, Federico
Ganguli, Surya
contents Large language Model (LLM)-assisted algorithm discovery is an iterative, black-box optimization process over programs to approximatively solve a target task, where an LLM proposes candidate programs and an external evaluator provides task feedback. Despite intense recent research on the topic and promising results, how can the LLM internal representation of the space of possible programs be maximally exploited to improve performance is an open question. Here, we introduce Contrastive Concept-Tree Search (CCTS), which extracts a hierarchical concept representation from the generated programs and learns a contrastive concept model that guides parent selection. By reweighting parents using a likelihood-ratio score between high- and low-performing solutions, CCTS biases search toward useful concept combinations and away from misleading ones, providing guidance through an explicit concept hierarchy rather than the algorithm lineage constructed by the LLM. We show that CCTS improves search efficiency over fitness-based baselines and produces interpretable, task-specific concept trees across a benchmark of open Erdős-type combinatorics problems. Our analysis indicates that the gains are driven largely by learning which concepts to avoid. We further validate these findings in a controlled synthetic algorithm-discovery environment, which reproduces qualitatively the search dynamics observed with the LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2602_03132
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Contrastive Concept-Tree Search for LLM-Assisted Algorithm Discovery
Leleu, Timothee
Gunathilaka, Sudeera
Ghimenti, Federico
Ganguli, Surya
Machine Learning
Artificial Intelligence
Neural and Evolutionary Computing
Large language Model (LLM)-assisted algorithm discovery is an iterative, black-box optimization process over programs to approximatively solve a target task, where an LLM proposes candidate programs and an external evaluator provides task feedback. Despite intense recent research on the topic and promising results, how can the LLM internal representation of the space of possible programs be maximally exploited to improve performance is an open question. Here, we introduce Contrastive Concept-Tree Search (CCTS), which extracts a hierarchical concept representation from the generated programs and learns a contrastive concept model that guides parent selection. By reweighting parents using a likelihood-ratio score between high- and low-performing solutions, CCTS biases search toward useful concept combinations and away from misleading ones, providing guidance through an explicit concept hierarchy rather than the algorithm lineage constructed by the LLM. We show that CCTS improves search efficiency over fitness-based baselines and produces interpretable, task-specific concept trees across a benchmark of open Erdős-type combinatorics problems. Our analysis indicates that the gains are driven largely by learning which concepts to avoid. We further validate these findings in a controlled synthetic algorithm-discovery environment, which reproduces qualitatively the search dynamics observed with the LLMs.
title Contrastive Concept-Tree Search for LLM-Assisted Algorithm Discovery
topic Machine Learning
Artificial Intelligence
Neural and Evolutionary Computing
url https://arxiv.org/abs/2602.03132