Machine Learning Dissertation Topics

Machine learning · Deep learning · Model evaluation

50 machine learning dissertation topics organised by method, from tree ensembles and imbalanced classification to CNNs, forecasting, clustering, interpretable AI, transfer learning and edge or federated systems.

50Research topics
10Method groups
13Data sources
1University repository
2016–26Thesis years

By Siddharth Gupta · Updated November 2026

A strong machine learning dissertation makes an evidence-based comparison, not just a complex model. A simple baseline, a leakage-safe validation strategy, appropriate performance metrics and reproducible data preparation often matter more than adding architecture complexity.

This method-led collection provides a question, proposed study design, suggested model comparison, starting dataset and related university thesis for each topic. The theses illustrate related research, not necessarily the same algorithm or dataset. Access ranges from open downloads to registration and approval.

Which machine learning method fits your data?

Your data or research problemStarting method family
Tabular, structured rows and columnsTree ensembles for tabular data
A rare event or anomaly classImbalanced data and anomaly detection
Images or visual classificationDeep learning for images
Clinical datasets and patient predictionMachine learning for health prediction
Signals, sequences or forecastsSequence models and forecasting
Unlabelled observationsClustering and representation learning
High-dimensional featuresFeature selection and dimensionality reduction
Need explanations, uncertainty or group fairnessInterpretability, uncertainty and fairness
Limited labelled imagesTransfer learning and generative models
Deployment efficiency, multiple sites or sequential decisionsEfficient, federated and reinforcement learning

How to use this list

  1. Match your available data to a method group below.
  2. Shortlist two or three ideas and inspect the dataset codebook, labels, sample size and usage licence.
  3. Define a transparent baseline and a fair train/validation/test split before model selection.
  4. Read the related thesis for context and reporting examples, without assuming it used the identical method.
  5. Confirm feasibility with your supervisor or book a free consultation.

Open means publicly accessible. Free login requires registration; Apply requires approval. Model training needs vary with dataset size and architecture.

Machine learning research area 01 / 10

Tree ensembles for tabular data

Most real data is a table. Random forests and gradient boosting are the strongest place to start.

02

Does cross-validation change which credit-risk model performs best?

Study design
Benchmark study
Method
Gradient boosting and baseline models with stratified cross-validation
Dataset
Home Credit Default Risk (Kaggle)

Feasibility note: The public Home Credit competition does not include a reliable application date for straightforward out-of-time validation. This study compares reproducible cross-validation strategies instead.

Related university thesis

Temporal Machine Learning for Predicting Accidents and Violations in the Mining IndustryNathan T. Kelley · University of Kentucky · 2025 · Related research example

03

Does gradient boosting still beat deep learning on tabular data?

Study design
Multi-dataset benchmark
Method
LightGBM compared with a tabular neural network
Dataset
OpenML Open
Related university thesis

Application Of Machine Learning in Transportation EngineeringAli Gorji Sefidmazgi · University of Kentucky · 2024 · Related research example

PDF details

Machine learning research area 02 / 10

Imbalanced data and anomaly detection

When the class you care about is rare, accuracy lies. Use precision, recall and the PR curve instead.

PDF details

Machine learning research area 03 / 10

Deep learning for images

CNN topics suit students with Python skills. Small benchmark sets train on a laptop.

PDF details

Machine learning research area 04 / 10

Machine learning for health prediction

Health ML topics need careful validation. Use patient-level splits and report calibration.

20

Can neural networks assign diagnosis codes from discharge notes?

Study design
Clinical text
Method
Multi-label neural text classifier
Dataset
MIMIC-IV (PhysioNet) Apply

Feasibility note: Diagnosis-code prediction from discharge notes additionally needs MIMIC-IV-Note, with separate access approval. Verify code-label alignment by admission.

Related university thesis

Deep Neural Networks for Multi-Label Text Classification: Application to Coding Electronic Medical RecordsAnthony Rios · University of Kentucky · 2018 · Related research example

PDF details

Machine learning research area 05 / 10

Sequence models and forecasting

Sequence topics use signals over time. Compare deep models with simple baselines before you claim a win.

24

Which window length works best for activity recognition?

Study design
Sensor data
Method
1D CNN with varied window sizes
Dataset
Human Activity Recognition Using Smartphones (UCI) Open

Feasibility note: This UCI dataset includes fixed-length pre-windowed sensor sequences. Testing arbitrary new window lengths may require original unsegmented recordings instead.

Related university thesis

Accelerometer-Based Vigilance State Classification in Dairy CowsEvan King · University of Kentucky · 2021 · Related research example

25

Do hand-crafted features or raw signals give better activity models?

Study design
Sensor data
Method
Random forest on features compared with deep learning on raw data
Dataset
Human Activity Recognition Using Smartphones (UCI) Open

Feasibility note: Compare the provided engineered feature tables with the linked inertial-signal files; keep all windows from each subject in the same validation split.

Related university thesis

Machine Learning Approach for Vigilance State Classification in MiceAnik Muhury · University of Kentucky · 2021 · Related research example

PDF details

Machine learning research area 06 / 10

Clustering and representation learning

Unsupervised topics find structure without labels. Always test whether clusters are stable.

PDF details

Machine learning research area 07 / 10

Feature selection and dimensionality reduction

With many features and few rows, selection matters more than the model. These topics test that.

PDF details

Machine learning research area 08 / 10

Interpretability, uncertainty and fairness

Examiners now ask why a model predicts what it does. These topics open the black box.

40

Do income models trained on one age group generalise to another?

Study design
Benchmark study
Method
Distribution shift test across age groups
Dataset
Adult Income (UCI) Open

Feasibility note: The UCI Adult dataset does not identify US states. Age-group transfer testing uses variables actually present in the dataset.

Related university thesis

Context Classification of Roadways Using Nationally Available GIS DataLaura Wright · University of Kentucky · 2021 · Related research example

PDF details

Machine learning research area 09 / 10

Transfer learning and generative models

Transfer learning lets you train good models on little data. Generative models create new data.

42

Do diffusion models make more varied images than GANs?

Study design
Image benchmark
Method
GAN compared with a diffusion model by FID
Dataset
Fashion-MNIST (Hugging Face) Open

Feasibility note: GANs and diffusion models can be computationally intensive. Use small models and reduced training budgets for a feasible student project.

Related university thesis

Advancing Generative Methods for Multimodal Data AnalysisRabeya Tus Sadia · University of Kentucky · 2026 · Related research example

PDF details

Machine learning research area 10 / 10

Efficient, federated and reinforcement learning

These topics study how ML runs in the real world, on devices, across sites and through trial and error.

46

Does federated learning match central training when data is split unevenly?

Study design
Image benchmark
Method
Federated averaging compared with central training
Dataset
Fashion-MNIST (Hugging Face) Open

Feasibility note: Simulated client partitions on Fashion-MNIST support an experimental federated benchmark; they do not represent real hospital or phone networks.

Related university thesis

Federated Data Engineering and Learning for Edge Intelligence SystemsAfsaneh Mahanipour · University of Kentucky · 2026 · Related research example

48

Can reinforcement learning set stock levels better than a simple rule?

Study design
Daily sales panel
Method
Deep Q-learning compared with a reorder rule
Dataset
M5 Forecasting (Kaggle, Walmart)

Feasibility note: M5 contains unit sales and prices, not observed replenishment decisions, stockouts or holding costs. Inventory-control claims must rely on explicitly stated simulation assumptions.

Related university thesis

A Comparative Analysis of Reinforcement Learning Applied to Task-Space Reaching With a Robotic Manipulator With and Without Gravity CompensationJonathan Fugal · University of Kentucky · 2020 · Related research example

50

Can a tiny model recognise activities on the phone itself?

Study design
Sensor data
Method
Compact CNN with latency testing
Dataset
Human Activity Recognition Using Smartphones (UCI) Open

Feasibility note: UCI sensor recordings support model-size and inference-speed benchmarking, but true phone latency requires testing on an actual device.

Related university thesis

3D Body Tracking using Deep LearningQingguo Xu · University of Kentucky · 2017 · Related research example

PDF details

Datasets for machine learning dissertations

Open datasets are publicly accessible, Free login requires an account and Apply requires approval. Some Kaggle competitions have licensing conditions; check them before research use.

OpenML Open

Thousands of ready-to-use machine learning datasets with standard tasks and benchmark results. Use it to compare algorithms across many datasets at once.

Fashion-MNIST (Hugging Face) Open

70,000 small greyscale images of clothing in 10 classes. Use it for image classification, autoencoder and generative model topics.

CIFAR-10 (University of Toronto) Open

60,000 colour images in 10 classes, the standard small image benchmark. Use it for CNN, transfer learning and model compression topics.

Home Credit Default Risk (Kaggle)

Loan applications with hundreds of features and a default label. Use it for gradient boosting, feature engineering and fairness topics.

Adult Income (UCI) Open

48,000 census records with income above or below 50,000 dollars. Use it for tree ensemble, interpretability and fairness topics.

NHANES (CDC) Open

US health survey with lab tests, body measures and diagnoses for thousands of adults. Use it for health prediction and clustering topics.

MIMIC-IV (PhysioNet) Apply

Hospital records with vitals, labs, notes and diagnosis codes for over 300,000 patients. Use it for clinical prediction and medical coding topics. Free-text notes require the separately approved MIMIC-IV-Note dataset.

Online Retail II (UCI) Open

Two years of transactions from a UK online retailer. Use it for customer clustering and embedding topics.

Printable machine learning dissertation guide

Get all 50 topics in one PDF

A printable list is being prepared. Meanwhile, all 50 questions, research designs, methods, datasets and related theses are available on this page.

PDF download coming soon

How to turn one of these into your own topic

  1. Start with a credible baseline. Use a simple model appropriate for the task, such as logistic regression for binary classification or seasonal naive forecasts for time series.
  2. Prevent data leakage. Split by time, person, device or organisation where observations are dependent, and keep the final test set untouched.
  3. Tune models fairly. Keep comparable computational budgets and use nested or separate validation for hyperparameter tuning.
  4. Use task-appropriate metrics. For rare classes consider PR-AUC, precision and recall; for forecasts use relevant error measures; for clustering assess stability and structure.
  5. Interpret performance carefully. Report uncertainty, subgroup errors, computational costs and practical limits alongside average scores.

Related dissertation topic collections

For broad project ideas, see the general data science collection. This page focuses specifically on machine learning methodology and benchmark design.

Need a second pair of eyes?

Choose a machine learning topic you can validate properly.

Bring your shortlist, data and planned evaluation strategy to a free 15-minute consultation about baselines, model choice, leakage and methodology.

Book a free 15-minute call

Ask about my machine learning analysis on WhatsApp

15 minutes · No cost · No pressure

Popular questions

What is a good machine learning dissertation topic?

A good topic answers a clear research question with accessible data, a relevant baseline and fair evaluation. Imbalanced classification, interpretability and transfer learning are useful areas to explore.

Do I need a powerful computer for a machine learning dissertation?

Not always. Many tabular benchmarks and small image classifiers run on a laptop or hosted notebook, but diffusion models, transformers, large hyperparameter searches and clinical datasets can require more computing or memory.

Which algorithms should I compare?

Choose a simple task-appropriate baseline, such as logistic regression or a decision tree for classification, and then compare a few stronger candidates such as random forest, gradient boosting or a neural network when justified.

How should I evaluate a machine learning model?

Use train/validation/test splits or cross-validation appropriate for the data. For imbalanced classification consider PR-AUC, precision, recall, F1 and calibration. Forecasting and clustering need different metrics.

Which software is best for machine learning?

Python with scikit-learn, XGBoost and PyTorch is widely used. R with tidymodels is also suitable for statistical learning and reproducible model evaluation.

Is there a machine learning topics PDF?

The printable Topic and Method list is being prepared. Every topic, method and dataset link is currently available on this page.

Perfect for students, researchers, and professionals looking to build real statistical skills.