News categorization based on titles with SVM, Naïve Bayesian, random forest, and RNN algorithms

Yongwei Li; Kejun Liu; Ziyu Liu; Zhen Tao; Meng Yuan

doi:10.1117/12.2641749

10 November 2022 News categorization based on titles with SVM, Naïve Bayesian, random forest, and RNN algorithms

Yongwei Li, Kejun Liu, Ziyu Liu, Zhen Tao, Meng Yuan

Proceedings Volume 12348, 2nd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2022); 123481D (2022) https://doi.org/10.1117/12.2641749
Event: 2nd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2022), 2022, Zhuhai, China

Abstract

News categorization, a text classification task, is now commonly used in many news websites. However, many of these news classifiers require full content of the news, which would cost great amounts of time for computation. In this paper, we focus on the possibility of categorizing news by its title with Support Vector Machines, Random Forest Classifiers, Naive Bayes, and Recurrent Neural Network. First, we explore some widely used pre-processing methods, including Bag of Words and Word2Vec. Then we combine these different pre-processing methods with the machine learning algorithms mentioned above to create different models. We measure their performances on the News Aggregator Data Set from UCI Machine Learning Repository, which contains over 400,000 pieces of news over 4 main categories. To evaluate the related performances, we use 85% data as a training set and 5% data as a validation set, and finally, use 10% data as a testing set. Comprehensive experimental results demonstrate that even with only the news titles, some models can still perform well in this challenging task. Therefore, it is possible to categorize news through its title in high accuracy yet with a much lower computing cost compared to full-text classification.

Citation Download Citation

Yongwei Li, Kejun Liu, Ziyu Liu, Zhen Tao, and Meng Yuan "News categorization based on titles with SVM, Naïve Bayesian, random forest, and RNN algorithms", Proc. SPIE 12348, 2nd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2022), 123481D (10 November 2022); https://doi.org/10.1117/12.2641749

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Data modeling

Statistical modeling

Neural networks

Performance modeling

Error analysis

Evolutionary algorithms

Machine learning

Show All Keywords

Keywords/Phrases

Search In:

Publication Years