• A
  • A
  • A
  • ABC
  • ABC
  • ABC
  • А
  • А
  • А
  • А
  • А
Regular version of the site

Researchers at HSE in St Petersburg Develop Superior Machine Learning Model for Determining Text Topics

Researchers at HSE in St Petersburg Develop Superior Machine Learning Model for Determining Text Topics

© iStock

They also revealed poor performance of neural networks on such tasks

Topic models are machine learning algorithms designed to analyse large text collections based on their topics. Scientists at HSE Campus in St Petersburg compared five topic models to determine which ones performed better. Two models, including GLDAW developed by the Laboratory for Social and Cognitive Informatics at HSE Campus in St Petersburg, made the lowest number of errors. The paper has been published in PeerJ Computer Science.

Determining the topic of a publication is usually not difficult for the human brain. For example, any editor can easily tag this article with science, artificial intelligence, and machine learning. However, the process of sorting information can be time-consuming for a person, which becomes critical when dealing with a large volume of data. A modern computer can perform this task much faster, but it requires solving a challenging problem: identifying the meaning of documents based on their content and categorising them accordingly.

This is achieved through topic modelling, a branch of machine learning that aims to categorise texts by topic. Topic modelling is used to facilitate information retrieval, analyse mass media, identify community topics in social networks, detect trends in scientific publications, and address various other tasks. For example, analysing financial news can accurately predict trading volumes on the stock exchange, which are significantly influenced by politicians' statements and economic events.

Here's how working with topic models typically unfolds: the algorithm takes a collection of text documents as input. At the output, each document is assessed for its degree of belonging to specific topics. These assessments are based on the frequency of word usage and the relationships between words and sentences. Thus, words such as ‘scientists,’ ‘laboratory,’ ‘analysis,’ ‘investigated,’ and ‘algorithms’ found in this text categorise it under the topic of ‘science.’

However, many words can appear in texts covering various topics. For example, the word ‘work’ is often used in texts about industrial production or the labour market. However, when used in the phrase ‘scientific work,’ it categorises the text as pertaining to ‘science.’ Such relationships, expressed mathematically through probability matrices, form the core of these algorithms.

Topic models can be enhanced by creating embeddings—fixed-length vectors that describe a specific entity based on various parameters. These embeddings serve as additional information acquired through training the model on millions of texts. 

Any phrase or text, such as this news item, can be represented as a sequence of numbers—a vector or a vector space. In machine learning, these numerical representations are referred to as embeddings. The idea is that measuring spaces and detecting similarities becomes easier, allowing comparisons between two or more texts. If the similarities between the embeddings describing the texts are significant, then they likely belong to the same category or cluster—a specific topic.

Scientists at the HSE Laboratory for Social and Cognitive Informatics in St Petersburg examined five topic models—ETM, GLDAW, GSM, WTM-GMM and W-LDA, which are based on different mathematical principles:

  • ETM is a model proposed by the prominent mathematician David M. Blei, who is one of the founders of the field of topic modelling in machine learning. His model is based on latent Dirichlet allocation and employs variational inference to calculate probability distributions, combined with embeddings.
  • Two models—GSM and WTM-GMM—are neural topic models.
  • W-LDA is based on Gibbs sampling and incorporates embeddings, but also uses latent Dirichlet allocation, similar to the Blei model.
  • GLDAW relies on a broader collection of embeddings to determine the association of words with topics.

For any topic model to perform effectively, it is crucial to determine the optimal number of categories or clusters into which the information should be divided. This is an additional challenge when tuning algorithms.

Sergei Koltsov

Sergey Koltsov, primary author of the paper, Leading Research Fellow, Laboratory of Social and Cognitive Informatics

Typically, a person does not know in advance how many topics are present in the information flow, so the task of determining the number of topics must be delegated to the machine. To accomplish this, we proposed measuring a certain amount of information as the inverse of chaos. If there is a lot of chaos, then there is little information, and vice versa. This allows for estimating the number of clusters, or in our case, topics associated with the dataset. We applied these principles in the GLDAW model.

The researchers investigated the models for stability (number of errors), coherence (establishing connections), and Renyi entropy (measuring the degree of chaos). The algorithms' performance was tested on three datasets: materials from a Russian-language news resource Lenta.ru and two English-language datasets - 20 Newsgroups and WoS. This choice was made because all texts in these sources were initially assigned tags, allowing for evaluation of the algorithms' performance in identifying the topics.

The experiment showed that ETM outperformed other models in terms of coherence on the Lenta.ru and 20 Newsgroups datasets, while GLDAW ranked first for the WoS dataset. Additionally, GLDAW exhibited the highest stability among the tested models, effectively determined the optimal number of topics, and performed well on shorter texts typical of social networks.

Sergey Koltsov, primary author of the paper, Leading Research Fellow, Laboratory of Social and Cognitive Informatics

We improved the GLDAW algorithm by incorporating a large collection of external embeddings derived from millions of documents. This enhancement enabled more accurate determination of semantic coherence between words and, consequently, more precise grouping of texts.

GSM, WTM-GMM and W-LDA demonstrated lower performance than ETM and GLDAW across all three measures. This finding surprised the researchers, as neural network models are generally considered superior to other types of models in many aspects of machine learning. The scientists have yet to determine the reasons for their poor performance in topic modelling.

See also:

Men Behind the Wheel: Three Times More Violations and Accidents than Women

Men are three times more likely than women to commit traffic violations while driving and to be involved in accidents. Moreover, they are more likely to create situations on the road that are highly dangerous to others. Men are also twice as likely to drive under the influence and nearly one-third more likely to receive a prison sentence for reckless driving. Perhaps it comes down to cultural norms and the different attitudes men and women have toward driving. These are the conclusions reached by Anton Kazun, Assistant Professor at the HSE Faculty of Economic Sciences, and Research Assistant Mikhail Belov.

HSE Scientists Discover How to Predict Charitable Behaviour Through Physiological Reactions

Researchers at the HSE Institute for Cognitive Neuroscience have investigated how the emotional impact of advertising affects the amount people willing to donate to support animal welfare. To accomplish this, the researchers measured physiological responses such as heart rate, electrodermal activity, and facial expressions in individuals viewing various photos of dogs. The findings indicate that willingness to donate is most accurately predicted by heart rate and facial muscle activation. The study has been published in Social Psychology. 

'We Are Creating the Medicine of the Future'

Dr Gerwin Schalk is a professor at Fudan University in Shanghai and a partner of the HSE Centre for Language and Brain within the framework of the strategic project 'Human Brain Resilience.' Dr Schalk is known as the creator of BCI2000, a non-commercial general-purpose brain-computer interface system. In this interview, he discusses modern neural interfaces, methods for post-stroke rehabilitation, a novel approach to neurosurgery, and shares his vision for the future of neurotechnology.

First Successful Attempt in 55 years: Physicists in Russia and Germany Confirm 1969 Experiment Results

A team of researchers, with the participation of physicists from HSE University, replicated the 1969 experiment on superconductivity and its properties. The scientists induced superconductivity by deliberately deteriorating the interfaces between the layers of superconductors and ferromagnets in the system, resulting in better performance of spin valves compared to the classical version, where the interfaces between the layers are ideal. This approach could lead to the development of more efficient devices for data storage and computing. The study findings have been published in the Beilstein Journal of Nanotechnology.

Healthy Nutrition Saves Public Funds: Strategies to Reduce Healthcare Costs in Russia

In Russia, the annual cost of treating type 2 diabetes alone exceeds 500 billion roubles. Promoting healthy nutrition programmes can ease the burden on the healthcare system and increase life expectancy. This was the conclusion reached by economists at HSE University after analysing global experiences with government involvement in promoting a healthy lifestyle.

Conscientious Individuals Live Longer

Personality traits such as conscientiousness, emotional stability, and an internal locus of control significantly influence one's lifestyle and longevity. Not only can personality traits influence health through beneficial and harmful habits but can also have a direct effect on mortality. Higher conscientiousness reduces the risk of premature death by 20 percentage points, while higher neuroticism increases it by 12 percentage points. These are the findings from a new study by Ksenia Rozhkova, Junior Research Fellow at the Laboratory for Labour Market Studies of the HSE Faculty of Economic Sciences.

Esports Players Play Better Online

In competitions, esports players, like other athletes, face stress and show worse results due to pressure. A substantial decrease takes place in the performance of esports players during overtime. This effect, however, is significantly mitigated in online competitions compared to live events—the difference can reach 30%. A study by a team of authors from HSE University’s Moscow and Perm campuses and European University Viadrina (Germany) explores the phenomenon of choking under pressure within the context of esports. The study was published in the Journal of Economic Behavior & Organization.

Analysing Genetic Information Can Help Prevent Complications after Myocardial Infarction

Researchers at HSE University have developed a machine learning (ML) model capable of predicting the risk of complications—major adverse cardiac events—in patients following a myocardial infarction. For the first time, the model incorporates genetic data, enabling a more accurate assessment of the risk of long-term complications. The study has been published in Frontiers in Medicine.

A New Tool Designed to Assess AI Ethics in Medicine Developed at HSE University

A team of researchers at the HSE AI Research Centre has created an index to evaluate the ethical standards of artificial intelligence (AI) systems used in medicine. This tool is designed to minimise potential risks and promote safer development and implementation of AI technologies in medical practice.

Smoking Habit Affects Response to False Feedback

A team of scientists at HSE University, in collaboration with the Institute of Higher Nervous Activity and Neurophysiology of the Russian Academy of Sciences, studied how people respond to deception when under stress and cognitive load. The study revealed that smoking habits interfere with performance on cognitive tasks involving memory and attention and impairs a person’s ability to detect deception. The study findings have been published in Frontiers in Neuroscience.