Skip to content

Latest commit

 

History

History
121 lines (94 loc) · 7.47 KB

README.md

File metadata and controls

121 lines (94 loc) · 7.47 KB
Raw

PoliticalLLM: Framework for Measuring and Manipulating Political Ideologies in LLMs

Wahl-O-Mat and PCT example

Code for the master's thesis “Steering Large Language Models towards Political Ideologies on Prompt-Level”. Demo is available here. This framework automatically evaluates the political ideology of LLMs with two ideology tests: Wahl-O-Mat and Political Compass Test. You can evaluate a base model or a manipulated model. Manipulation is done through Prompt Enginnering and RAG. The model can be directly steered towards a target German political party (impersonation) or indirectly (RAG with most similar contexts from manifesto database or random ideological triggers from same database). For the RAG model, German manifesto data from the Manifesto Project is embedded and saved in a Chroma vector database.

Quickstart

Create a new venv (Python 3.9.18) and clone the repo. (Note: Make sure to have enough disk space as the script downloads a sentence embedding model)

git clone https://github.com/j0st/PoliticalLLM

Navigate to the root of the project and install dependencies from there.

pip install -r requirements.txt

To run this project with API models, you will need to add the following environment variables to your .env file. Create this file in the root directory of the project. By default, OpenAI, together.ai and Anyscale API endpoints are integrated.

# .env file

# Models
OPENAI_API_KEY=""

ANYSCALE_API_KEY=""
ANYSCALE_BASE_URL="https://api.endpoints.anyscale.com/v1"

TOGETHER_AI_API_KEY=""
TOGETHER_AI_BASE_URL="https://api.together.xyz/v1"

# Set this if you want to use your own llama.cpp model locally
LOCAL_LLAMA_MODEL_PATH=""

# Data
MANIFESTO_PROJECT_API_KEY=""

Testing can be done in a new Python file or in the existing main.py. After importing the LLM class from this project, you can create an instance with the desired LLM and call the ideology test methods. Parameters values and explanations can be found below.

# main.py

from llms import LLM

ChatGPT = LLM("gpt-3.5-turbo-0125")

ChatGPT.wahlomat(filename="YOUR_FILENAME", plot_results=True)
ChatGPT.pct(filename="YOUR_FILENAME", plot_results=True)

After finishing the tests, the following files are created in the results folder:

  • responses-YOUR_FILENAME.csv - Lists all (mapped) responses from the LLM to each political statement.
  • descriptives-YOUR_FILENAME.csv - Descriptive stats for each statement answered by the LLM.
  • plot-YOUR_FILENAME.png - Plot of the results.

Project Structure

.
├── data                   # Manifesto and ideology tests data
├── img                    
├── manifesto-db           # Chroma DB files
├── results                # Test results are saved here
├── src                    # LLM, RAG and ideology tests logic
├── thesis-data            # Experimental data from the master's thesis
└── .gitattributes
└── .gitignore
└── .README.md
└── __init__.py
└── requirements.txt

Project Files Description

  • src/main.py - Used to run experiments with LLMs.
  • src/llms.py - Base class for implementing LLMs and ideology tests as methods.
    • $~$
  • src/tests/pct.py - Selenium script to run the PCT test.
  • src/tests/wahlomat.py - Calculates the agreement scores between parties.
    • $~$
  • src/rag/retriever.py - Retrieves top k statement from manifesto database (RAG).

Some other supporting files

  • src/map_answers.py - Maps the answers from an LLM to the ideology tests (e.g. "Agree") to int values which are needed to do the ideology test.
    • $~$
  • src/analysis/descriptives.py - Calculates the mean, median, mode and std in the list of responses provided after iterating through the statements.
  • src/analysis/pct_plot_spectrum.py - Plots the PCT coordinates on a two-dimensional spectrum.
  • src/analysis/wahlomat_radar_chart.py - Plots the Wahl-O-Mat agreements scores between parties on a radar chart.
    • $~$
  • src/rag/embeddings/chunking.py - Chunks manifesto data for embedding model.
  • src/rag/embeddings/embeddings.py - Creates Chroma.db embeddings from manifesto dataset.
  • src/rag/embeddings/synthetic_dataset.py - Generates a synthetic QA pair dataset.
  • src/rag/embeddings/evaluation.py - Evaluates embedding model against validation synthetic dataset.
  • src/rag/embeddings/fine_tuning.ipynb - Script for fine-tuning embedding model.
    • $~$
  • data/scripts/manifesto_project.py - Get manifesto data from Manifesto Project API.

.pct() and .wahlomat() parameters

  • filename (str): Specifies the filename for saving the results, which includes CSV files with responses, descriptive statistics, and a PNG image showing placement on the two-dimensional spectrum (pct) or on the radar chart (wahlomat).
  • party (Optional[str]): When set, the prompt is modified to impersonate the given political party. Default is None.
  • ideology (Optional[str]): Works only in conjunction with rag=True. It restricts retrieved context to a specific ideology. Possible values include "Authoritarian-right", "Authoritarian-left", "Libertarian-left", "Libertarian-right". Default is None.
  • n_results (Optional[int]): Applicable when rag=True. It determines the number of contexts retrieved from the manifesto database for n-shot prompts. Default is None.
  • rag (bool): Enables the Retrieval Augmented Generation pipeline, inserting retrieved contexts from a vector database into the prompt. Default is False.
  • rag_mode (Optional[str]): Specifies the mode of operation for the RAG pipeline. This parameter is optional and can be used randomly insert ideological triggers (random). Default is similarity.
  • plot_result (Optional[bool]): If True, the results are plotted and saved in a specified data folder. Default is False. Please note that Selenium is used to calculate the coordinates of the PCT. It might throw exceptions due to unexected browser changes. In these cases, the script tries again until successful.
  • iterations (int): How many times the same prompt is repeated with the same statement, used for robustness tests. Default is 1.

Supported Models

More Models

You can easily add your own models or change the API provider in the query method of the LLM class in llms.py. Just make sure that the query method returns the response in a string.

Demo

You can try a demo of the different prompt manipulations for different LLMs on huggingface.co/spaces/jost/PoliticalLLM. Note that Hugging Face Spaces need to restart after some inactivity time which takes a few minutes.