Extreme Learning Machines for Fast Training of Click-Through Rate Prediction Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autor principal: Biçici, Ergun
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914849121894400
author Biçici, Ergun
author_facet Biçici, Ergun
contents Extreme Learning Machines (ELM) provide a fast alternative to traditional gradient-based learning in neural networks, offering rapid training and robust generalization capabilities. Its theoretical basis shows its universal approximation capability. We explore the application of ELMs for the task of Click-Through Rate (CTR) prediction, which is largely unexplored by ELMs due to the high dimensionality of the problem. We introduce an ELM-based model enhanced with embedding layers to improve the performance on CTR tasks, which is a novel addition to the field. Experimental results on benchmark datasets, including Avazu and Criteo, demonstrate that our proposed ELM with embeddings achieves competitive F1 results while significantly reducing training time compared to state-of-the-art models such as Masknet. Our findings show that ELMs can be useful for CTR prediction, especially when fast training is needed.
format Preprint
id arxiv_https___arxiv_org_abs_2406_17828
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Extreme Learning Machines for Fast Training of Click-Through Rate Prediction Models
Biçici, Ergun
Machine Learning
Artificial Intelligence
I.2; I.5.1
Extreme Learning Machines (ELM) provide a fast alternative to traditional gradient-based learning in neural networks, offering rapid training and robust generalization capabilities. Its theoretical basis shows its universal approximation capability. We explore the application of ELMs for the task of Click-Through Rate (CTR) prediction, which is largely unexplored by ELMs due to the high dimensionality of the problem. We introduce an ELM-based model enhanced with embedding layers to improve the performance on CTR tasks, which is a novel addition to the field. Experimental results on benchmark datasets, including Avazu and Criteo, demonstrate that our proposed ELM with embeddings achieves competitive F1 results while significantly reducing training time compared to state-of-the-art models such as Masknet. Our findings show that ELMs can be useful for CTR prediction, especially when fast training is needed.
title Extreme Learning Machines for Fast Training of Click-Through Rate Prediction Models
topic Machine Learning
Artificial Intelligence
I.2; I.5.1
url https://arxiv.org/abs/2406.17828