IEEE BigData 2024 Cup — Predicting Chess Puzzle Difficulty

Machine Learning Difficulty Estimation Combining Neural Chess Engines
👥 Team of 3
📅 November 2024
Technologies & Stack
Python Scikit-learn LightGBM / XGBoost Chess Engines (Stockfish, Leela, Maia) Feature Engineering Pandas / NumPy
IEEE BigData 2024 Cup — Predicting Chess Puzzle Difficulty

Project Overview

Participated in the IEEE BigData 2024 Cup competition with team PlatsBruts, competing to accurately predict the human solving difficulty rating (Elo) of tactical chess puzzles from millions of positions.

Tactical Chess Puzzle Difficulty Prediction Figure 1: Tactical chess puzzle position evaluation combining deep engine search depth (Stockfish) with human behavioral probability distributions (Maia Chess).

Predicting puzzle difficulty is challenging because traditional chess engines evaluate objective positional advantage rather than human cognitive difficulty. A move that is trivial for an engine to calculate can be virtually invisible to a human master, and vice versa.


Technical Methodology & Feature Engineering

1. Engine Heuristic Features

  • Stockfish Search Dynamics: Evaluated positions at varying depths to extract centipawn volatility, move order stability, node count expansion, and evaluation deltas between the best move and second-best alternative.
  • Shannon Entropy of Move Distribution: Calculated the entropy over candidate move probabilities; high entropy reflects ambiguous positions with many plausible-looking decoy moves (blunder traps).

2. Human Behavior Modeling with Maia Chess

  • Maia Neural Networks: Unlike standard superhuman engines, Maia is trained directly on millions of human games across specific rating tiers (Maia 1100 to Maia 1900) to predict the exact move a human at that rating would make.
  • Extracted disagreement rates between superhuman ground truth (Stockfish) and human expectation (Maia) as direct proxies for tactical difficulty and intuitive traps.

3. Gradient Boosted Decision Ensembles

  • Trained LightGBM and XGBoost regression models with Bayesian hyperparameter optimization and k-fold cross-validation, optimizing Root Mean Squared Error (RMSE) against verified human rating benchmarks.