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

RAG-pipeline Optimization: HSE Data Case

Student: Berestova Viktoria

Supervisor: Armen Beklaryan

Faculty: Faculty of Computer Science

Educational Programme: Financial Technology and Data Analysis (Master)

Year of Graduation: 2024

This work explores a method of accessing large language models, which involves retrieval-augmented generation (RAG). The study builds all the stages of the RAG pipeline and optimizes them using data from the Higher School of Economics (HSE). The results of the work will be applied to develop a chatbot assistant for HSE University students. RAG is built on the Russian large language model GigaChat using the model's API through the GigaChain SDK. Various methods of text processing, vector databases, prompting techniques, and large language model generation methods were employed in the study. The final result of the work is an automated hyperparameters tuning pipeline for a RAG built on a given knowledge base. As part of the research, a parser of the knowledge database based on relevant sources about the educational process at the University was also developed, a set of validation metrics for assessing the quality of answers was formed, and the RAG automatic validation framework (RAGAS) was used. The analysis carried out allowed us to draw conclusions about the influence of each stage of the pipeline on the final answers. As a result of the work, the main metrics of response quality were significantly improved (+30 percentage points in response correctness). The research also identified shortcomings of the current approach, namely the low quality of the initial data. The findings will be used for further research and development of the chat assistant within the project "Using Large Language Models to Assist HSE Students." Directions for improvement and growth points of the work were formulated.

Student Theses at HSE must be completed in accordance with the University Rules and regulations specified by each educational programme.

Summaries of all theses must be published and made freely available on the HSE website.

The full text of a thesis can be published in open access on the HSE website only if the authoring student (copyright holder) agrees, or, if the thesis was written by a team of students, if all the co-authors (copyright holders) agree. After a thesis is published on the HSE website, it obtains the status of an online publication.

Student theses are objects of copyright and their use is subject to limitations in accordance with the Russian Federation’s law on intellectual property.

In the event that a thesis is quoted or otherwise used, reference to the author’s name and the source of quotation is required.

Search all student theses