• Stars
    star
    470
  • Rank 90,147 (Top 2 %)
  • Language
  • Created about 1 year ago
  • Updated about 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

The release of the Twitter algorithm, annotated for recsys

Awesome Twitter Algo ๐Ÿฆ

Curated by Igor Brigadir and Vicki Boykis.

An annotated look through the release of the Twitter algorithm, through the context of engineering and recsys, with notes from repo creators on significance of specific parts of the code. Since it can be hard to parse through so much code and derive meaning and context, we do it for you!

This code focuses on the services used to build the Home timeline For You feed, the algorithmic tab that is now served first on both web and mobile next to the Following feed.

Screenshot 2023-03-31 at 9 36 04 PM

Contributing

We're happy to take changes that add and contextualize Twitter's recommendations algorithm as it's been released over the past week. To contribute, please submit a PR with good formatting and grammar and lots of links to references where relevant. We're especially happy for feedback from tweeps or former tweeps who can tell us where we got it wrong.

High-level Context and Summary

One thing that's immediately obvious is that this is not the entire codebase or even a working majority of it. Missing from this codebase are

  1. many flows that process,enrich, and refine model input data
  2. YAML configuration metafiles which could tell us quite a bit about how the code actually works. There are only 7 of them, the rest have been redacted.
  3. Most code related to spinning up the actual infrastructure
  4. Git commit history that shows us how some of this code has evolved
  5. A large portion of the trust and safety codebase, which Twitter has noted they've left out for now

An important high-level concept discussed in the Spaces releasing this code was in-network and out-of-network. In-network tweets are those from people you follow, out-of-network is everyone else. A blend of 50%/50% are offered in the daily ~1500 tweets run through rankers.

Code Links

What was released? The majority of the code and algorithms, but not the data or parameters or configurations or build tools of the Recommender Systems behind "For You" timeline recommendations. The Candidate Retrieval code was also not released, and neither was the Trust and Safety components, and the Ads components - those remain closed off. No User Data or credentials were inside the repositories and code comments were sanitized (or at least, none were obviously there on first look).

Twitter Algo Repo || Twitter ML Algo Repo || Blog Post

Recsys Architecture Diagram

twitter architecture@2x (1)

Link to update here.

Twitter Data-Centric Historical Architecture Context

There is a very, very old post from 2013 on High Scalability which gives some context to how these systems were initially constructed.

As context, Twitter initially ran all workloads on-prem but has been moving to Google Cloud.. In 2019, Twitter began by migrating to BigQuery and DataFlow from a data and analytics perspective. Before the move to BigQuery, much of the data was stored in HDFS using Thrift. It currently lives in BigQuery and is processed for many of the pipelines described below using DataFlow, GCP's Spark/Scalding-processing equivalent platform.

Programming Languages and Frameworks

The released code comes in a variety of languages. The most common languages used at Twitter are:

Java

  • Used in Lucene for search indexing
  • Used in the GraphJet library

Scala

  • Scalding, written at Twitter, pre-cursor to Spark
  • Scio for parallel cluster computing computations

Python

  • Python for the machine learning models in the stack, includes both PyTorch and Tensorflow (legacy) code

Frameworks and Metalanguages

Internal Libraries

  • Finagle is a service written in Scala with Java and Scala APIs used to manage RPCs
  • Snowflake is a service that generates unique identifiers for each tweet based on timestamp, worker number, and sequence number
  • Heron - A realtime streaming analytics library (similar to Flink)
  • Strato - a virtual database powered by microservices

Recsys

recsys

The typical recommender system pipeline has four steps: candidate generation, ranking, filtering, and serving. Twitter has many pipelines for performing verious parts of this this across the overall released codebase.

  • Candidate generation occurs when you have millions or billions of potential items in your source data based on user-item interactions. This piece usually includes collaborative filtering or neural algorithms to reduce the size of the candiate dataset for downstream tasks.
  • These need to then be ranked against each other, filtered against business logic and blended and served to the user in a given surface area, in this case the For You feed in the Twitter timeline.

Input Data

The input data comes from:

  • Kafka
  • Twitter Eventbus
  • GCS
  • Vertica
  • Manhattan, a real-time multitenant distributed database that was initially developed as a serving layer on top of Hadoop and includes both observability and other metrics.

Screenshot 2023-04-01 at 3 26 24 PM

In migrating to GCP, the current data ingest looks something like this:

  • Streaming Dataflow jobs to apply deduping
  • Perform real-time aggregation and sink data into BigTable

That data is then made available to the candidate generation phase. There is not much about the actual data, even what a schema might look like, in the repo.

Candidate Generators

(also called "features" in the chart)

GraphJet


  • GraphJet - A realtime Java graph processing library that allows for in-memory processing on a single server and focuses on providing content recommendations. Paper here. Recommendations are provided based on shared interests, correlated activities, and a number of other input signals. GraphJet maintains a realtime bipartite interaction graph that keeps track of userโ€“tweet interactions over the most recent n hours and reads from Kafka. Each individual GraphJet server can ingest one million graph edges per second and compute 500 recommendations/second.

They describe the reasons specifically for creating an in-memory DB in the GraphJet paper:

In terms of recommendation algorithms, we have found that random walks, particularly over bipartite graphs, work well for generating high-engagement recommendations. Although conceptually simple, random-walk algorithms define a large design space that supports customization for a wide range of application scenarios, for recommendations in different contexts (web, mobile, email digests, etc.) as well as entirely unrelated applications (e.g., social search). The output of our random-walk algorithms can serve as input to machine-learned models that further increase the quality of recommendations, but in many cases, the output is sufficiently relevant for direct user consumption.

In terms of production infrastructure for generating graph recommendations, the deployed systems at Twitter have always gone "against the grain" of conventional wisdom. When many in the community were focused on building distributed graph stores, we built a solution (circa 2010) based on retaining the entire graph in memory on a single machine (i.e., no partitioning). This unorthodox design decision enabled Twitter to rapidly develop and deploy a missing feature in the service (see Section 2.1). Later, when there was much activity in the space of graph processing frameworks rushing to replace MapReduce, we abandoned the in-memory system and reimplemented our algorithms in Hadoop MapReduce (circa 2012). Once again, this might seem like another strange design decision (see Section 2.2). Most recently, we have supplemented Hadoop-based recommendations with custom infrastructure, first with a system called MagicRecs (see Section 2.3) and culminating in GraphJet, the focus of this paper.

The precursor to GraphJet was WTF, Who to Follow, which focused only on recommending users to other users., using Cassovary, an in-memory graph processing engine built specifically for WTF, also built on the JVM.

Screenshot 2023-04-02 at 12 22 15 PM

GraphJet implements two random walk algorithms:

  • Circle of Trust (internal to Twitter) and
  • SALSA (Stochastic Approach for Link-Structure Analysis).

Screenshot 2023-04-02 at 12 38 16 PM

GraphJet Architecture
  • A large portion of the traffic to GraphJet comes from clients who request content recommendations for a particular user.

GraphJet includes CLICK, FAVORITE, RETWEET, REPLY, AND TWEET as input node types and keeps track of left (input) and right(output) nodes.

SimClusters


SimClusters is another recommended tweet candidate generation source that, given an embedding (or a model-learned vector representation of an entity, in this case a tweet, event, or topic), will return a group of candidate tweets (or events, or topics) that are similar to the input content via approximate nearest neighbors lookup using approximate cosine similarity as a distance metric.

SimClusters are built using this algorithm and served using the ANN service.

SimClusters Embeddings Algorithm

See Paper here

There is a multitude of content that can be recommended on Twitter: Tweets, user recommendations, events, hashtags, and Who to Follow, the service discussed in the GraphJet and Who to Follow papers.

Each of these recommendation algorithms need to be relearned frequently since content on Twitter moves quickly, and they need to be presented in a variety of places: not only in the feed, but also via email or notifications, or the "Trends and Events" section.

However, the speed of change for different features is different: for example user recommendations change much more slowly than topic and tweet recommendations. Tweet embeddings are critical for tweet recommendation tasks. We can calculate tweet similarity and recommend similar tweets to users based on their tweet engagement history.

Screenshot 2023-04-04 at 6 36 16 AM

Previous iterations of homogenous recommendations for specific domains included WTF and GraphJet, but each hard their limitations. For example, GraphJet can generate a bipartite graph for recommendations at real time, but does not generalize to new domain areas.

We can generalize all of these representations and learn them all from a single service, which was the idea behind SimClusters.

In SimClusters, we don't use traditional recsys methods (i.e. matrix factorization) because it's computationally expensive in this case. Instead, we base it on ANN and construct a user-user cluster graph based on community structures (i.e. K-Pop or machine learning) where the central figures in the community are "influencers", and content is recommended on a per-community level.

The algorithm for community detection is based on Metropolis-Hastings, computed offline on MapReduce jobs in Hadoop.

It can compute clusters for 1 bil users (note that Twitter has 450 million MAU -monthly active users, which means that the clusters are computed on all potential users? ) and 100k dimensions, with each dimension representing a specific community, which allows it to represent long-tail content fairly well.

Simclusters Engineering Implementation

First, we ingest input data that includes tweets, topics, etc. It's unclear where this data comes from.

Stage 1: Looking at the user-user graph and creating a list of communities that users belong to - "User Interest Representations" This is run as an MR job in Hadoop.

Stage 2: Calculates the representations for a given target, given a user-user bipartite graph, running in parallel.

Screenshot 2023-04-04 at 6 44 52 AM

Both of these stages can be run independently.

Embeddings are generated and can be extended to work in batch-distributed, batch-multicore, or streaming-distributed modes.

They are then used downstream and blended with sources in the CR mixer.

There is also a realtime component, SimClusters ANN, which can return similar content based on the SimClusters output embedding. A Heron job builds the mapping between SimClusters and Tweets. The job saves top 400 Tweets for a SimClusters and top 100 SimClusters for a Tweet.

Filters

If a user opted out of an interest group or category, they're ignored from the SimClusters index.

Hypothetical Architecture Diagram

Simclusters


TwHIN

TwHIN Code,TwHIN Paper

Twitter's Heterogeneous Information Network (HIN) is a graph network where the nodes/vertices of the graph represent multiple entity types, and the edges represent one of many interaction types between the entities.

The following entity and relation types are represented:

  • Entity Types (Nodes): User, Tweet, Advertiser, Ad
  • Relation Types (Edges): Follow, Authors, Favourites, Replies, Retweets, Promotes, Clicks

The multi-type, multi-relation network enables the resultant TwHIN embeddings to capture signals such as:

  • social signals (follow-graph)
  • content engagement signals (tweet, image, and video engagements)
  • advertisement engagements

TwHIN at Twitter

The TwHIN approach is applied to form two different heterogenous networks, each centered around a high coverage relation:

1. TwHIN-Follow

kNN-Embed paper, Huggingface Dataset

A Twitter HIN centered around a User-User follow graph. Graph of User (consumer) to Author (producer) nodes, where an edge represents a user "following" an author engagement

  • 261M edges and 15.5M vertices
  • Max-degree of 900๐พ and a min-degree of 5.

2.TwHIN-Engagement

MiCRO paper, Huggingface Dataset

A Twitter HIN centered around a User-Tweet engagement graph. Graph of User to Tweet nodes, where an edge represents a "fave" engagement

  • 6.7M user nodes, 13M Tweet nodes, and 283M edges
  • For users: max-degree of 100 and min-degree of 1
  • For tweets: max-degree of 280k and min-degree of 5.

Each of the TwHIN datasets released on Huggingface have been heavily subsampled, and anonymized due to privacy restrictions.

Training

  • local.yaml defines some training configurations
  • machines.yaml defines the resources for training the TWHIN embeddings. Notably,ย it specifies 16x A100 GPUs & 1.4TB of RAM.

RealGraph

RealGraph is a pipeline that aims to predict the top K users that a given user would interact with given some historical interaction features. The ETL portions are implemented in scio, while the ML training/scoring portions are all done in BigQuery. Given the huge space of users O(1e9) and to avoid the huge O(N**2) complexity of storing all interactions between every possible pair of users, Twitter reduces the amount of data processed by adopting the following methods:

  • Getting only users from recent follows, direct interactions (likes, profile clicks), and push notification open events (code). Note that while the folder mentions labels, it is used as a set of positive examples to generate the candidates, as can be seen from this BQ query.
  • Using only a very narrow set of 10 highly informative features here.
  • Scoring is done with a xgboost model trained in BQML, and scored in BQML (BQ one-liner).

For users that have no explicit activity (eg dormant or new users), the candidates are backfilled using the top k tweeting accounts that the given user follows (BQ query). This is important to ensure that Twitter is still able to send notifications/emails to dormant users, or even to make sure they have something ready if and when the dormant user returns.


TweepCred


Earlybird

Mixers

Before candidate tweets are sent to be ranked in the light and heavy rankers before being presented to the user, they are combined from their various candidate generation sources within several entities, one being the CR Mixer, which fetches out-of-network recommended candidate tweets.

CR Mixer

Rankers

Based on the Twitter engineering blog post, a total of 1500 candidates are retrieved. However, only some of them will be served to your Twitter feed.

Twitter would want to show the tweets that you are most likely to positively engage with. Therefore Twitter will predict probabilities of whether you will engage with the tweet, and use these probabilities to score the tweets.

To reduce computation cost, tweets are first ranked with a light ranker (which is just a logistic regression) and then a heavy ranking (a neural network model).

Light Ranker

This is their documentation

  • โ€œThe Earlybird light ranker is a which predicts the likelihood that the user will engage with a tweet. It is intended to be a simplified version of the heavy ranker which can run on a greater amount of tweets.โ€
  • โ€œThe current model was last trained several years ago, and uses some very strange features. We are working on training a new model, and eventually rebuilding this part of the stack entirely.โ€

Twitter has separate models for ranking in-network and out-network tweets, with different features

The model for the Light Ranker TensorFlow model is trained using TWML which is said to be deprecated, but the code is in deepbird project.

The Earlybird Light Ranker has some feature weights but as suggested in the code, they are read in as run time parameters and these are most likely different in practice.

Heavy Ranker

It's important to note that there are no content-based embeddings inside the main ranking algorithm.

Input features. All the specific features within the input feature list are based slightly on content signals and mostly social signals, such as "aggregate counts of user interaction with other engagers of tweets that the user interacts with", and based heavily on "likes" and "replies" as input actions, but at an aggregate level. The social and embeddings-based features in the dataset are not used and weighted as much.

All of these are combined and weighted into a score. Hyperparameters for the model and weighting are here.

For more details on the model, see the Architecture overview.

Scoring Plan

After the model predicts the probability of the actions, weights are assigned to the probability. The tweet with the highest score is likely to appear at the top of your feed.

These are the actions predicted, and their corresponding weights

feature weight
probability the user will favorite the Tweet (0.5)
probability the user will click into the conversation of this tweet and reply or like a Tweet (11*)
probability the user will click into the conversation of this Tweet and stay there for at least 2 minutes (11*)
probability the user will react negatively (requesting "show less often" on the Tweet or author, block or mute the Tweet author) (-74)
probability the user opens the Tweet author profile and Likes or replies to a Tweet (12)
probability the user replies to the Tweet (27)
probability the user replies to the Tweet and this reply is engaged by the Tweet author (75)
probability the user will click Report Tweet (-369)
probability the user will ReTweet the Tweet (1)
probability (for a video Tweet) that the user will watch at least half of the video (0.005)

The score of the tweet is equal to

P(favorite) * 0.5 + max( P(click and reply), P(click and stay two minutes) ) * 11 + P(hide or block or mute) * -74 + ... etc

The tweet with the highest score is likely to appear at the top of your feed. (There is still a part on boost where multipliers will be applied to the score). However, filtering is applied afterwards, and this could change what tweets you actually see.

There are some interpretations we can make from the scoring plan

  • They combine the negative feedback actions (hide/mute/block) even though they have different produce consequences. By combining the predictions I think they hope to generalize the signal. However, the report prediction is by itself and has a much larger negative weight.
  • There is very limited implicit action in the scoring plan. This is unlike short video recommendation systems like TikTok where the system learns from how long you stay on the video. The weight for the video completion prediction is insignificant.
  • The only implicit action being predicted is when you click into the conversation of this Tweet and stay there for at least 2 minutes. 2 minutes is quite a large number. This can be viewed as a defense against comment bait, where the author entices you to click on the comments but leave you disappointed. If you exit the comment section soon after clicking, it is not considered a positive signal to engagement.
  • The scoring plan encourages participation in the conversation. The weight for the probability of you replying is high. The weight for the probability of the author replying to your reply is even higher. We can view this as Twitter's intention to be the "town square" of the Internet. However, this signal does not differentiate whether the conversation is friendly or otherwise (unless you also hide/mute/block/report).
  • We should also note that the score of Blue Verified authors will be given a multiplier of 4 or 2, which overrides many of the weights in the scoring plan.

The release does not describe how the weights are chosen. We expect the weights to be tuned with A/B testing. We are also curious about what Twitter measures and optimizes when they tune the weights.

Filters

Usually, filtering happens before ranking to avoid the need to rank candidates that will be filtered later. However, on Twitter, the blog implies that filtering happens after ranking.

  • visibility-filters

    • (From the blog) "Filter out Tweets based on their content and your preferences. For instance, remove Tweets from accounts you block or mute."
    • โ€œVisibility Filtering library is currently being reviewed and rebuilt, and part of the code has been removed and is not ready to be shared yet. The remaining part of the code needs further review and will be shared once itโ€™s ready. Also code comments have been sanitized.โ€
  • Remove out-of-network competitor site URLs from potential offered candidate Tweets

Ordering

There are some reasons why we might not want to order the tweets strictly by the scoring plan. The scoring plan scores tweets independent of other Tweets. However, we might want to consider other tweets when presenting the tweets on the feed, for example, avoid showing tweets from the same author consecutively or maintain some other form of diversity in the tweets.

These are the heuristics mentioned in the blog

  • Author Diversity: Avoid too many consecutive Tweets from a single author.

    • See code
    • score * ((1 - 0.25) * Math.pow(0.5, position) + 0.25)
    • If you have seen the author is the same feed refresh, the score of the tweet from the author halved (but with a floor)
  • Content Balance: Ensure we are delivering a fair balance of In-Network and Out-of-Network Tweets.

    • (Contributions needed)
  • Feedback-based Fatigue: Lower the score of certain Tweets if the viewer has provided negative feedback around it.

    • See code
    • The multiplier will be less than one if you
      • Provided negative feedback on the author of the tweet
      • Provided negative feedback to the users who like the tweet
      • Provided negative feedback on users who follow the author of the tweet (?)
      • Provided negative feedback on users who retweeted the tweet
    • Recent negative feedback will have a greater weight
      • If the negative feedback is provided more than 14 + 140 days ago, the negative feedback will not be considered.
      • If the negative feedback was provided less than 14 days ago, the tweet will be filtered. See code
  • Social Proof: Exclude Out-of-Network Tweets without a second degree connection to the Tweet as a quality safeguard. In other words, ensure someone you follow engaged with the Tweet or follows the Tweetโ€™s author.

    • What is described above is a filter, not a discount. However, we can find the discount, see code
    • ScaleFactor = 0.75 is applied to out-of-network tweets (exactly second degree connection?), in-network retweets of out-of-network tweets should not have this multiplier applied
    • We might have a filter that removes all content with more than two degrees of connection.
  • Twitter Blue boost: (Not listed in blog)

    • See code and default parameters
      • If the author of the candidate tweet is a Blue Verified and in the network of the user (i.e. user follows author?), the score of the tweet is multiplied by 4
      • If the author of the candidate tweet is a Blue Verified and out of the network of the user (i.e. does not follow author an within two degrees of connection), the score of the tweet from is multiplied by 2.
    • This means that Blue Verified authors that the user does not follow is given a greater boost than the authors the user explicitly follows.
    • Note that Twitter Blue is launched shortly after Elon Musk's takeover.

Business Terms and Logic

These are Twitter specific terms and names that keep coming up across different code bases and blog posts.

  • Twepoch - A "magic number" 1288834974657L, which is a timestamp for 2010-11-04T01:42:54Z the date that Twitter introduced the Snowflake ID system, used as Twitter's own "Unix Epoch"
  • Snowflake - Twitter's system for assigning unique IDs to tweets, users, lists, DMs, media etc.
  • WTF - Who to follow
  • DDG - Duck Duck Goose, Twitter's A/B Testing Platform.
  • Earlybird - Twitter's Lucene based real-time search index. Notes and a blog post here.
  • "Unregretted user minutes" - the metric Twitter publicly states is the thing they are optimizing for. It is unknown how exactly they measure this.

Bias and Manipulation

Cases of potential bias, manipulation, favouritism, hacks, etc. The focus on this repository is on the concrete, technical aspects of the code, not speculating on anything twitter may or may not have done. That exercise is left to the reader, however, there are some technical aspects that should still be described about these popular accusations, this is a section for those. Unfortunately, much of the configuration that would contain specific instances of interventions is not in the code.

Deboosting Rival Sites

It was long speculated youtube links get massively deboosted, and Spaces links massively boost Tweets in recommendations. There are no specific references to this in the code. However, there are filters that could be configured for this, referencing OutOfNetworkCompetitorURLFilter for example.

Elon Musk feature

The Elon Musk / Democrat / Republican Code: Now Removed. One of the first widely shared cases, falsely assuming this is something that directly affects Recommendations when it was actually for internal A/B testing, to monitor for effects (DDG is Duck Duck Goose, the A/B Testing Platfom). It was also mentioned in the space and denied there. However, a former twitter employee also offered an alternative explanation (A/B Testing measures behavior, so one way or another Twitter is tuning your TL, indirectly).

Ukraine

There are two mentions related to Ukraine in the Twitter Algo repo. Whereas one of them is a flag for Ukraine-related misinformation used for moderation, or warning labels, there is another safety label for Twitter Spaces called UkraineCrisisTopic. Here are some facts about these labels and their function:

Changes

  • 2 hours after it was released, Twitter removed feature flags that specifically highlighted Elon's account

Discussions about The Algorithm Elsewhere

Resources for Learning More about Recsys

More Repositories

1

twitter-advanced-search

Advanced Search for Twitter.
1,163
star
2

stopwords

Default English stopword lists from many different sources
Python
280
star
3

DownloadConceptualCaptions

Reliably download millions of images efficiently
Jupyter Notebook
107
star
4

ishkurs-guide-dataset

Structured Data from Ishkur's Guide to Electronic Music. Working Mirror for v2.5 here: https://igorbrigadir.github.io/ishkurs-guide-dataset/
Jupyter Notebook
43
star
5

twitter-history

Tracking significant changes to the Twitter API or platform as a whole
19
star
6

covid19-twitter-stream-tool

A tool to ingest the Twitter COVID-19 Labs Stream
Python
8
star
7

insight-templates

LaTeX Templates for Insight Centre for Data Analytics
TeX
7
star
8

word2vec-java

word2vec-java
Java
7
star
9

docker-spacy-gpu

Minimal example of a GPU Docker container that runs SpaCy Transformers
Dockerfile
7
star
10

newsir16-data

Additional External Data for Signal Media One-Million News Articles Dataset used in NewsIR 16 ECIR Workshop
Jupyter Notebook
6
star
11

twitter-glossary

A glossary of Twitter specific terminology.
5
star
12

awesome-bluesky-algo

The Bluesky algorithm, annotated for recsys. (Joking.. unless??)
5
star
13

bluesky-top-ua

Top Ukrainian Users on Bluesky
HTML
2
star
14

igorbrigadir

Profile
2
star
15

simetrix

Mavenized Fork of SIMetrix by Annie Louis
Java
2
star
16

twitter-ads-transparency

Data from the Twitter Ads Transparency Center https://ads.twitter.com/transparency
Python
2
star
17

carp

๐ŸŸ Twitter Carp Data
2
star
18

kaggle-word2vec

https://www.kaggle.com/c/word2vec-nlp-tutorial/
Python
2
star
19

SemEval2014-Task1

Distributional Semantic Model Tests for SemEval Task 1 Relatedness Subtask
Python
1
star
20

ROUGE-BEwTE

Mavenized Fork of "BEwT-E: Basic Elements with Transformations for Automated Evaluation of Summaries"
Java
1
star
21

Common-Recommender-REST-API

OpenAPI Spec for the Common Recommender REST API
1
star
22

tweet-delete

Makefile
1
star
23

power100-redux

Re Ranking Irish #power100 List
1
star
24

wiki-docs-submodule

Github wiki as git submodule
1
star
25

Count-von-Count

Count-vector-based distributional semantic approaches
Shell
1
star